def test_removal_of_a_module_removes_its_code_tree(self): project = EmptyProject() mod = project.create_module('module.py') ct = CodeTree(None) project.remember_code_tree(ct, mod) project.remove_module(mod.subpath) assert_raises(CodeTreeNotFound, lambda: CodeTree.of(mod))