def test_init_stratum_should_append_point_to_strata_group(self):
pmap = EarthMatrix([[]])
pmap._strata = [['strata_group_0'], ['strata_group_1']]
point = EarthPoint(1, 5, 15)
pmap._init_stratum(point)
self.assertEqual([point], pmap._strata[2])
def test_init_stratum_should_assign_next_free_strata_group(self):
pmap = EarthMatrix([[]])
pmap._strata = [['strata_group_0'], ['strata_group_1']]
point = EarthPoint(1, 5, 15)
pmap._init_stratum(point)
self.assertEqual(2, point.stratum)