def test_class_relationship(xl, workbook):
CODE = '''
class Person(object):
@vbmeta(name=String, age=Integer)
def __init__(self, name, age):
self.name = name
self.age = age
class Company(object):
@vbmeta(employees=Collection)
def __init__(self, employees):
self.employees = employees
@vbmeta(rettype=String)
def test():
c = Company([
Person("James", 27),
Person("Bob", 22)
])
return c.employees[1].name
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'test', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
def test_basic_function(xl, workbook):
CODE = '''
@vbmeta(x=Integer, y=Integer, rettype=Integer)
def add(x, y):
return x + y'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'add', globals(), xl, workbook)
pyresult = pyfcn(5,4)
vbaresult = vbafcn(5,4)
assert pyresult == vbaresult
def test_dictionary_literal(xl, workbook):
CODE = '''
@vbmeta(rettype=Integer)
def dicttest():
d = {'hello' : 1, 'world' : 2}
return d['hello']
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'dicttest', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
def test_integer_for_loop(xl, workbook):
CODE = '''
def test():
sum = 0
for i in range(1,10):
sum += i
return sum
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'test', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
def test_array_literal(xl, workbook):
CODE = '''
@vbmeta(rettype=Integer)
def arraytest():
a = [1,2,3,4.4]
return a[0] + a[2]
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'arraytest', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
def test_function_call(xl, workbook):
CODE = '''
@vbmeta(x=Integer, y=Integer, rettype=Integer)
def add(x, y):
return x + y
@vbmeta(x=Integer, y=Integer, rettype=Integer)
def sumsquare(x, y):
return add(x*x, y*y)
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'sumsquare', globals(), xl, workbook)
pyresult = pyfcn(5,4)
vbaresult = vbafcn(5,4)
assert pyresult == vbaresult
def test_elif_statement(xl, workbook):
CODE = '''
def sign(x):
if x < 0:
return -1
elif x == 0:
return 0
else:
return 1
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'sign', globals(), xl, workbook)
cases = [-5, 0, 5]
for case in cases:
pyresult = pyfcn(case)
vbaresult = vbafcn(case)
assert pyresult == vbaresult
def test_basic_class(xl, workbook):
CODE = '''
class TestClass(object):
@vbmeta(name=String)
def __init__(self, name):
self.name = name
@vbmeta(rettype=String)
def test():
t = TestClass('James')
return t.name
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'test', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
def test_if_statement(xl, workbook):
CODE = '''
@vbmeta(x=Integer, y=Integer, rettype=Integer)
def max(x, y):
if x > y:
return x
else:
return y
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'max', globals(), xl, workbook)
pyresult1 = pyfcn(1,5)
vbaresult1 = vbafcn(1,5)
pyresult2 = pyfcn(5,1)
vbaresult2 = vbafcn(5,1)
assert pyresult1 == vbaresult1
assert pyresult2 == vbaresult2
def test_nested_if_statement(xl, workbook):
CODE = '''
def absadd(x, y):
if x < 0:
if y < 0:
return -x-y
else:
return -x+y
else:
if y < 0:
return x-y
else:
return x+y
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'absadd', globals(), xl, workbook)
cases = [(-5,5),(-5,-5),(5,5),(5,-5)]
for case in cases:
pyresult = pyfcn(*case)
vbaresult = vbafcn(*case)
assert pyresult == vbaresult
def test_list_comprehension(xl, workbook):
CODE = '''
@vbmeta(rettype=Collection)
def test():
s = 2
x = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
y = [s*z for z in x if z > 3 and z <= 9 and z % 2 == 0]
return y
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'test', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
pycount = len(pyresult)
vbacount = vbaresult.Count()
assert pycount == vbacount
for i in range(pycount):
pyobj = pyresult[i]
vbaobj = vbaresult.Item(i+1)
assert pyobj == vbaobj
def test_array_and_dict_assignments(xl, workbook):
CODE = '''
@vbmeta(rettype=Integer)
def add(x, y):
return x + y
@vbmeta(rettype=Integer)
def sumsquare(x, y):
return add(x*x, y*y)
@vbmeta(rettype=Integer)
def complicated_assignments():
x = [{'name' : 'James', 'age' : 27},
{'name' : 'Bob', 'age' : 22}]
return sumsquare(x[0]['age'], x[1]['age'])
'''
pyfcn, vbafcn = lift_code_to_py_and_vba_functions(CODE, 'complicated_assignments', globals(), xl, workbook)
pyresult = pyfcn()
vbaresult = vbafcn()
assert pyresult == vbaresult
