def test_sources(self): srcs = ["compiler.py", "propgcc.py"] compiler = Compiler(files=srcs) self.assert_equal(len(srcs), len(compiler.get_files())) def gen_sources(): return ["cc.py", "python.py"] compiler.add_file(gen_sources) self.assert_equal(len(compiler.get_files()), len(srcs) + len(gen_sources()))