def test_pickle(self) -> None:
"""test if generic code objects are pickleable"""
code = generic.GenericCode([
"1",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("test", None, origin="this test", is_package=False))
code2 = pickle.loads(pickle.dumps(code))
self.assertEqual(code2.sections, code.sections)
self.assertEqual(code2.spec, code.spec)
def test_build(self) -> None:
"""test the building of a generic code object"""
builder = generic.GenericCodeBuilder(-1)
builder.add_text("1", 0)
builder.add_code("numbers.append('2')", 0)
builder.add_code("numbers.append('3')", 0)
builder.add_text("4", 0)
spec = ModuleSpec("test", None, origin="this test", is_package=False)
code = builder.code(spec)
code2 = generic.GenericCode(
("1", compile("numbers.append('2')", "this test", "exec"),
compile("numbers.append('3')", "this test", "exec"), "4"), spec)
self.assertEqual(code, code2)
def test_equal(self) -> None:
"""test if equality between generic code objetcs works"""
code = generic.GenericCode([
"1",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("test", None, origin="this test", is_package=False))
code2 = generic.GenericCode([
"1",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("test", None, origin="this test", is_package=False))
code3 = generic.GenericCode([
"X",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("test", None, origin="this test", is_package=False))
code4 = generic.GenericCode([
"1",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("X", None, origin="this test", is_package=False))
self.assertEqual(code, code2)
self.assertNotEqual(code, code3)
self.assertNotEqual(code, code4)
self.assertNotEqual(code3, code4)
def test_execute(self) -> None:
"""test the execution"""
code = generic.GenericCode([
"1",
"2",
compile("numbers.append('3')", "<string>", "exec"),
"4",
compile("numbers.append('5')", "<string>", "exec"),
], ModuleSpec("__main__", None, origin=None, is_package=False))
numbers = []
variables = {"numbers": numbers}
for number in code.execute(variables):
numbers.append(number)
self.assertEqual(numbers, ["1", "2", "3", "4", "5"])
def test_constants(self) -> None:
"""test the module level constants"""
spec = ModuleSpec("test", None, origin="this test", is_package=False)
code = generic.GenericCode([], spec)
variables = {}
list(code.execute(variables)) # modifies variables
self.assertEqual(
variables, {
"__name__": "test",
"__loader__": None,
"__file__": "this test",
"__path__": None,
"__package__": "",
"__cached__": None,
"__spec__": spec
})