def test_search(self): """Test search by searching existing route""" train = Train.objects.all()[0] st1 = train.trainpath_set.all()[0].station st2 = train.trainpath_set.all()[1].station # from st1 to st1 search = search_train_raw(st1.city.name, st2.city.name, None) self.assertTrue(train in list(search)) # but not backwards search = search_train_raw(st2.city.name, st1.city.name, None) self.assertFalse(train in list(search))
def test_date_search(self): """Test search by searching existing route""" train = Train.objects.all()[0] tp1 = train.trainpath_set.all()[0] st1, time1 = tp1.station, tp1.time st2 = train.trainpath_set.all()[1].station search = search_train_raw(st1.city.name, st2.city.name, None) for train in search: st = train.trainpath_set.get(station=st1) self.assertTrue(abs(st.time - time1) <= timezone.timedelta(1))