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_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_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_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_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)
def test_export_transfers_empty(self):
transfers_txt = Transfer.export_txt(self.feed)
self.assertFalse(transfers_txt)
