def test_remove(self): top = Assembly() top._setup() g = top._depgraph.component_graph() comps = [name for name in g] self.assertEqual(comps, ['driver']) top.add('comp', Component()) top._setup() g = top._depgraph.component_graph() comps = [name for name in g] self.assertEqual(set(comps), set(['driver', 'comp'])) top.remove('comp') top._setup() g = top._depgraph.component_graph() comps = [name for name in g] self.assertEqual(comps, ['driver'])