def test_add_coord_with_coord_none(self): station = factories.StationFactory(longitude=None, latitude=None) city = station.city station = add_coord_to_station_by_name(city.name_cn, station.name_cn, '10.10', '12.321') self.assertEqual(station.longitude, Decimal('10.10')) self.assertEqual(station.latitude, Decimal('12.321'))
def test_add_coord_with_coord_none(self): station = factories.StationFactory(longitude=None, latitude=None) city = station.city station = add_coord_to_station_by_name(city.name_cn, station.name_cn, '10.10', '12.321') self.assertEqual(station.longitude, Decimal('10.10')) self.assertEqual(station.latitude, Decimal('12.321'))
def test_add_coord_with_existing_coord(self): station = factories.StationFactory(latitude=None) city = station.city old_lng = station.longitude station = add_coord_to_station_by_name( city.name_cn, station.name_cn, old_lng + Decimal('1'), '12.321' ) self.assertEqual(station.longitude, old_lng) self.assertEqual(station.latitude, Decimal('12.321')) old_lng = station.longitude old_lat = station.latitude station = add_coord_to_station_by_name( city.name_cn, station.name_cn, old_lng + Decimal('1'), old_lat + Decimal('1'), override=True ) self.assertEqual(station.longitude, old_lng + Decimal('1')) self.assertEqual(station.latitude, old_lat + Decimal('1'))
def test_add_coord_with_existing_coord(self): station = factories.StationFactory(latitude=None) city = station.city old_lng = station.longitude station = add_coord_to_station_by_name(city.name_cn, station.name_cn, old_lng + Decimal('1'), '12.321') self.assertEqual(station.longitude, old_lng) self.assertEqual(station.latitude, Decimal('12.321')) old_lng = station.longitude old_lat = station.latitude station = add_coord_to_station_by_name(city.name_cn, station.name_cn, old_lng + Decimal('1'), old_lat + Decimal('1'), override=True) self.assertEqual(station.longitude, old_lng + Decimal('1')) self.assertEqual(station.latitude, old_lat + Decimal('1'))