test_memory.py

from openalea.lpy import *

#def test_String( nb = 1, size = 100):
#    """ Test creation of str """
#    for i in xrange(nb):
#        a = 'F'*size
#        del a

#maxsize = 10000
#maxlength = 8
maxsize = 100
maxlength = 3


def test_AT( nb = maxlength, size = maxsize):
    """ Test creation of AxialTree """
    for i in range(nb):
        a = AxialTree('F'*size)
        del a

def test_PAT( nb = maxlength, size = maxsize):
    """ Test creation of Parametric AxialTree """
    for i in range(nb):
        a = AxialTree()
        for j in range(size):
            a += 'F('+str(j)+')'
        del a

def test_LsRuleWithGlobalContext():
    """ Test creation of LsysRule with global context """
    l = LsysRule()
    try:
        l.set('F --> F')
    except NameError as e :
        import warnings
        warnings.warn("GlobalContext has not lpy symbols")

def test_LsRule():
    """ Test creation of LsysRule """
    lc = LsysContext()
    lc.makeCurrent()
    res = 'F[+F]F[-F]F'
    l = LsysRule()
    l.set('F --> '+res)
    assert l.isCompiled(), "LsysRule.compile did not work."
    assert l.forwardCompatible() == True and l.backwardCompatible() == True 
    assert l.isContextFree() == True and l.nbContexts() == 0
    assert l.nbParameters() == 0
    #assert len(l()) == len(res)
    assert len(l.process('FF')) == 2*len(res)
    assert l.match('F',0)[0] == 1
    assert l.reverse_match('FFF')[0] == 1

def lnLs(l = 8):
    if l == 0: return 1
    elif l == 1: return 11
    else:
        res = 6
        a = 5
        for i in range(l-2):
            res += a*6
            a *= 5
        res += a*11
        return res
        
def test_Ls( nb = 1, length = maxlength):
    """ Test creation and execution of Lsystems """
    l = Lsystem()
    l.derivationLength = length
    l.axiom = AxialTree('F')
    l.addProductionRule('F --> F[+F]F[-F]F',0)
    ln = lnLs(length)
    for i in range(nb):
        a = l.iterate()
        assert len(a) == ln
        del a
    del l

def test_PLs( nb = 1, length = maxlength):
    """ Test creation and execution of Parametric Lsystems """
    l = Lsystem()
    l.derivationLength = length
    l.axiom = AxialTree('F(0)')
    l.addProductionRule('F(x) --> F(x+1)[+F(x)]F(x+1)[-F(x)]F(x+1)',0)
    ln = lnLs(length)
    for i in range(nb):
        a = l.iterate()
        assert len(a) == ln
        del a
    del l

if __name__ == '__main__':
    test_AT()
    test_PAT()
    test_LsRuleWithGlobalContext()
    test_LsRule()
    test_Ls()
    test_PLs()