def test_enterprise_route_count(self):
"""Test that the enterprise service returns the correct number of routes
"""
service = Service('enterprise')
with self.assertRaises(NotImplementedError):
service.routes()
def test_metro_route_count(self):
"""Test that the metro service returns the correct number of routes
"""
service = Service('metro')
self.assertEqual(102, len(service.routes()))
def test_ulsterbus_route_count(self):
"""Test that the ulsterbus service returns the correct number of routes
"""
service = Service('ulsterbus')
self.assertEqual(454, len(service.routes()))
def test_goldline_route_count(self):
"""Test that the goldline service returns the correct number of routes
"""
service = Service('goldline')
self.assertEqual(34, len(service.routes()))
