def test_saves_year_correctly(self):
vertex = factories.VertexFactory.create()
vertex.year = "2050"
vertices = {2: vertex}
reach = factories.ReachFactory.create(slug="MA")
riversegment = factories.RiverSegmentFactory(
location=1, reach=reach)
import_helpers.import_vertex_row(vertices, [1.0, "MA", 5])
vertex_value = models.VertexValue.objects.get(
riversegment=riversegment,
vertex=vertex,
year="2050")
self.assertEquals(vertex_value.value, 5)
def test_saves_column_after_empty_column(self):
vertex1 = factories.VertexFactory.create()
vertex2 = factories.VertexFactory.create()
vertex1.year = "2050"
vertex2.year = "2050"
vertices = {2: vertex1, 3: vertex1, 4: vertex2}
reach = factories.ReachFactory.create(slug="MA")
riversegment = factories.RiverSegmentFactory(
location=1, reach=reach)
import_helpers.import_vertex_row(vertices, [1.0, 'MA', 1.0, '', 2.0])
value1 = models.VertexValue.objects.get(
riversegment=riversegment, vertex=vertex1, year="2050")
value2 = models.VertexValue.objects.get(
riversegment=riversegment, vertex=vertex2, year="2050")
self.assertEquals(value1.value, 1.0)
self.assertEquals(value2.value, 2.0)
