def test_none(self): inp = ["1", "2"] for graph, lines in _itergraph("foo", inp): self.assertEqual(graph, "foo") self.assertEqual(["1", "2"], list(lines))
def test_two(self): inp = ["multigraph foo_1", "1", "2", "multigraph foo_2", "3", "4"] for graph, lines in _itergraph("foo", inp): if graph == "foo_1": self.assertEqual(["1", "2"], list(lines)) elif graph == "foo_2": self.assertEqual(["3", "4"], list(lines)) else: self.fail()