def test_with_random():
import pytest
with pytest.raises(Exception) as info:
transform("""
def test_oops():
with foo as bar:
pass
""")
assert str(info.value) == \
"Cannot handle with blocks that aren't pytest exception checks"
def test_if_constant():
import pytest
with pytest.raises(Exception) as info:
transform("""
def test_oops():
if True:
pass
""")
assert str(info.value) == \
"Cannot handle ifs that aren't is_pyd or is_pynih"
def test_assign_to_var():
ir = transform("""
def test_assign():
foo = 42
""")
assert ir == [
AutowrapTest("test_assign", [Assignment('foo', NumLiteral(42))])
]
def test_if_pyd():
ir = transform("""
def test_if_pyd():
if is_pyd:
pass
""")
assert ir == [AutowrapTest("test_if_pyd", [
IfPyd([]),
])]
def test_if_python():
ir = transform("""
def test_if_python():
if is_python:
pass
""")
assert ir == [AutowrapTest("test_if_python", [
IfPython([]),
])]
def test_assign_from_bytes():
ir = transform("""
def test_assign_from_bytes():
foo = b'abc'
""")
assert ir == [
AutowrapTest("test_assign_from_bytes",
[Assignment('foo', BytesLiteral(b'abc'))])
]
def test_assign_to_list():
ir = transform("""
def test_assign_to_list():
[foo, bar] = 42
""")
assert ir == [
AutowrapTest("test_assign_to_list",
[Assignment(Sequence(['foo', 'bar']), NumLiteral(42))])
]
def test_if_pynih():
ir = transform("""
def test_if_pynih():
if is_pynih:
pass
""")
assert ir == [AutowrapTest("test_if_pynih", [
IfPynih([]),
])]
def test_with_raises_exception():
ir = transform("""
def test_raises():
with pytest.raises(ImportError):
from bar import foo
""")
assert ir == [
AutowrapTest("test_raises",
[ShouldThrow("ImportError", [Import('bar', ['foo'])])])
]
def test_import_from():
ir = transform("""
def test_import_from():
from foo import bar, baz
""")
assert ir == [
AutowrapTest("test_import_from", [
Import("foo", ["bar", "baz"]),
])
]
def test_attribute():
ir = transform("""
def test_func():
foo.value = 42
""")
assert ir == [
AutowrapTest("test_func",
[Assignment(
Attribute('foo', 'value'),
NumLiteral(42),
)])
]
def test_one_assertion_literals_1():
ir = transform("""
def random_function():
pass
def test_bar():
assert 3 == 'foo'
""")
assert ir == [
AutowrapTest("test_bar", [
Assertion(NumLiteral(3), StringLiteral("foo")),
])
]
def test_one_assertion_literals_0():
ir = transform("""
def random_function():
pass
def test_foo():
assert 1 == 2
""")
assert ir == [
AutowrapTest("test_foo", [
Assertion(NumLiteral(1), NumLiteral(2)),
])
]
def test_tuple():
ir = transform("""
def test_tuple():
assert (1, 2) == (3, 4, 5)
""")
assert ir == [
AutowrapTest("test_tuple", [
Assertion(
Sequence([NumLiteral(1), NumLiteral(2)]),
Sequence([NumLiteral(3),
NumLiteral(4),
NumLiteral(5)]),
)
])
]
def test_assertion_call_function():
ir = transform("""
def test_func():
from mod import func
assert func(7) == 42
""")
assert ir == [
AutowrapTest("test_func", [
Import('mod', ['func']),
Assertion(
FunctionCall("func", [NumLiteral(7)]),
NumLiteral(42),
),
])
]
def test_import():
ir = transform("""
def test_import():
import foo, bar
a = foo.GLOBAL
""")
assert ir == [
AutowrapTest("test_import", [
Import("foo", []),
Import("bar", []),
Assignment(
'a',
Attribute('foo', 'GLOBAL'),
)
])
]
def test_assertion_call_method_on_lvalue():
ir = transform("""
def test_func():
obj.func(42, 3.3)
""")
assert ir == [
AutowrapTest("test_func", [
FunctionCall(
Attribute(
'obj',
'func',
),
[NumLiteral(42), NumLiteral(3.3)],
),
])
]
def test_subscript():
ir = transform("""
def test_subscript():
arr = [1, 2, 3]
fst = arr[1]
""")
assert ir == [
AutowrapTest("test_subscript", [
Assignment(
'arr', Sequence([
NumLiteral(1),
NumLiteral(2),
NumLiteral(3),
])),
Assignment('fst', Index('arr', NumLiteral(1)))
])
]
def test_assertion_call_method_on_rvalue():
ir = transform("""
def test_func():
from mod import Klass
Klass('ctor_arg').func(42, 3.3)
""")
assert ir == [
AutowrapTest("test_func", [
Import('mod', ['Klass']),
FunctionCall(
Attribute(
FunctionCall('Klass', [StringLiteral('ctor_arg')]),
'func',
),
[NumLiteral(42), NumLiteral(3.3)],
),
])
]
def test_length():
ir = transform("""
def test_length():
lst = [1]
len(lst)
""")
assert ir == [
AutowrapTest(
"test_length",
[
Assignment('lst', Sequence([NumLiteral(1)])),
FunctionCall(
Attribute(
'lst',
'length',
),
[], # args
)
])
]
def translate(source_code, filename):
from python_to_ir import transform
tests = transform(source_code)
with open(filename, "w") as file:
writer = Writer(file)
writer.writeln("// this file is autogenerated, do not modify by hand")
with NamedBlock(writer, "",
f"namespace Autowrap.CSharp.Tests") as block:
_write_imports(block, tests)
# we use the fully-qualified names to avoid name-collisions
# with the symbols from the test
with NamedBlock(
block,
"[Microsoft.VisualStudio.TestTools.UnitTesting.TestClass]",
"public class TestMain") as block:
for test in tests:
_translate_test(block, test)
