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_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()
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()
