def test_export_transfers_minimal(self): Transfer.objects.create( from_stop=self.stop1, to_stop=self.stop2) transfers_txt = Transfer.export_txt(self.feed) self.assertEqual(transfers_txt, """\ from_stop_id,to_stop_id,transfer_type STOP1,STOP2,0 """)
def test_export_transfers_maximal(self): Transfer.objects.create( from_stop=self.stop1, to_stop=self.stop2, transfer_type=2, min_transfer_time=5) transfers_txt = Transfer.export_txt(self.feed) self.assertEqual(transfers_txt, """\ from_stop_id,to_stop_id,transfer_type,min_transfer_time STOP1,STOP2,2,5 """)
def test_export_transfers_empty(self): transfers_txt = Transfer.export_txt(self.feed) self.assertFalse(transfers_txt)