def test_addPackage(self): saved = modulegraph._packagePathMap self.assertIsInstance(saved, dict) try: modulegraph._packagePathMap = {} modulegraph.addPackagePath('foo', 'a') self.assertEqual(modulegraph._packagePathMap, { 'foo': ['a'] }) modulegraph.addPackagePath('foo', 'b') self.assertEqual(modulegraph._packagePathMap, { 'foo': ['a', 'b'] }) modulegraph.addPackagePath('bar', 'b') self.assertEqual(modulegraph._packagePathMap, { 'foo': ['a', 'b'], 'bar': ['b'] }) finally: modulegraph._packagePathMap = saved