malar
/
iloads-cyclicity
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added tests

master
Malar Kannan 2017-07-03 17:19:51 +05:30
parent 9f836cbbb9
commit 081b957b45
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cyclic_contracts.py
View File

@ -1,4 +1,4 @@
from cyclic_utils import CYCLE_LIMIT,rem_routes from cyclic_utils import rem_routes
class Route(object): class Route(object):
"""docstring for Route.""" """docstring for Route."""
@ -67,7 +67,8 @@ class Transporter(object):
if len(unfulfilled) > 0: if len(unfulfilled) > 0:
print "Contracts from ",repr(unfulfilled) print "Contracts from ",repr(unfulfilled)
def contract_corridors(self):
return rem_routes(self.contract_routes())
if __name__ == '__main__':
# Transporter.from_file('./contracts.txt').contracts_required() Transporter.from_file('./contracts.txt').contracts_required()
Transporter.from_file('./contracts.txt').contracts_required()

12
cyclic_contracts_test.py
View File

@ -1,14 +1,18 @@
import unittest import unittest
import cyclic_contracts from cyclic_contracts import Transporter
class CyclicTest(unittest.TestCase): class CyclicTest(unittest.TestCase):
"""docstring for CyclicTest.""" """docstring for CyclicTest."""
def test_loads(): def test_contract(self):
cc = Transporter.from_file("./contracts.txt").contract_corridors()
self.assertEqual(cc, [('Mumbai', 'Bangalore')])
def test_partial(self):
cc = Transporter.from_file(
"./contracts_partial.txt").contract_corridors()
self.assertEqual(cc, [('Mumbai', 'Kolkata'), ('Chennai', 'Kolkata')])
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()