Esempio n. 1
0
    def test_preprocessor_unique(self):
        pp = IdentifierPreprocessor()
        pp.compile("foo(a,a,b,b,c,c,c,c) + foo(a,b,c)")

        functions = sorted(list(f.name for f in pp.functions))
        variables = sorted(list(v.name for v in pp.variables))

        self.assertEqual(functions, ["foo"])
        self.assertEqual(variables, ["a", "b", "c"])
Esempio n. 2
0
    def test_preprocessor(self):
        pp = IdentifierPreprocessor()
        pp.compile("foo(a + b) * bar(b + c)")

        functions = set(f.name for f in pp.functions)
        variables = set(v.name for v in pp.variables)

        self.assertEqual(functions, set(["foo", "bar"]))
        self.assertEqual(variables, set(["a", "b", "c"]))