test_predecessor_at_scale.py

from openalea.lpy import *

lsysbasecode = '''
module U,E : scale = 1 
module I,L : scale = 2 
Axiom: 
  nproduce E L U I I I 
  for i in range(1):
    nproduce U I I I
  nproduce E L
'''

res_1 = [None, 0, 0, 2, 2, 2, 2, 6, 6, 6, 6, 10]
res_2 = [None, None, 1, 1, 3, 4, 5, 5, 7, 8, 9, 9]


def i_predecessor_at_scale(i):
    l = Lsystem()
    l.setCode(lsysbasecode)
    l.makeCurrent()
    lstring = l.axiom
    m = lstring[i]
    pred = lstring.predecessor_at_scale(i,1)
    pred2 = lstring.predecessor_at_scale(i,2)
    res = i,m,pred,(lstring[pred] if not pred is None else pred),pred2,(lstring[pred2] if not pred2 is None else pred2)
    l.done()
    return res



def test_predecessor_at_scale(assertion = True):
    l = Lsystem()
    l.setCode(lsysbasecode)
    l.makeCurrent()
    lstring = l.axiom
    print(lstring)
    for i,m in enumerate(lstring):
        pred = lstring.predecessor_at_scale(i,1)
        pred2 = lstring.predecessor_at_scale(i,2)
        print(i,m, end=' ')
        print('\t',pred, end=' ')
        if not pred is None: print(lstring[pred], end=' ')
        print('\t',pred == res_1[i], end=' ')
        print('\t',pred2, end=' ')
        if not pred2 is None: print(lstring[pred2], end=' ')
        print('\t',pred2 == res_2[i])
        if assertion:
            assert pred == res_1[i] and pred2 == res_2[i]

if __name__ == '__main__':
    test_predecessor_at_scale(False)