def insert_graph(self):
with open(self.filename, 'rb') as lgl_file:
lgl = lgl_file.read()
for i, entry in enumerate(lgl.split(os.linesep)):
if not entry:
continue
if self.max_num_of_vertices != -1 and i > self.max_num_of_vertices:
break
if LGLReader.is_starting_vertex(entry):
self._process_primary_vertex(entry, i)
else:
self._process_secondary_vertex(entry, i)
def test_given_start_vertex_returns_true(self):
self.assertTrue(LGLReader.is_starting_vertex('# the_vertex'))
def test_given_non_start_vertex_returns_false(self):
self.assertFalse(LGLReader.is_starting_vertex(''))
self.assertFalse(LGLReader.is_starting_vertex(' '))
self.assertFalse(LGLReader.is_starting_vertex('one one'))
