def test_import_transfers_txt_duplicate(self): transfers_txt = StringIO("""\ from_stop_id,to_stop_id STOP1,STOP2 STOP1,STOP2 """) Transfer.import_txt(transfers_txt, self.feed) transfer = Transfer.objects.get() # Just one self.assertEqual(transfer.from_stop, self.stop1) self.assertEqual(transfer.to_stop, self.stop2)
def test_import_transfers_txt_omitted(self): transfers_txt = StringIO("""\ from_stop_id,to_stop_id,transfer_type,min_transfer_time STOP1,STOP2,, """) Transfer.import_txt(transfers_txt, self.feed) transfer = Transfer.objects.get() self.assertEqual(transfer.from_stop, self.stop1) self.assertEqual(transfer.to_stop, self.stop2) self.assertEqual(transfer.transfer_type, 0) self.assertEqual(transfer.min_transfer_time, None)
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)