def test_header_no_columns_works_correctly(self): row_values = ["Some name"] vertices = import_helpers.build_vertex_dict(row_values) vertex = vertices[2] self.assertEquals(vertex.header, '') self.assertEquals(vertex.name, 'Some name') self.assertEquals(vertex.year, '2100')
def test_header_year_and_header_works_correctly(self): row_values = ["2050: Whee: Some name"] vertices = import_helpers.build_vertex_dict(row_values) vertex = vertices[2] self.assertEquals(vertex.header, 'Whee') self.assertEquals(vertex.name, 'Some name') self.assertEquals(vertex.year, '2050')
def test_build_vertex_dict_creates_vertices(self): row_values = ["Some name", "Some other name"] import_helpers.build_vertex_dict(row_values) self.assertEquals(models.Vertex.objects.count(), 2)