def test_more_than_distance(self):
number_of_trips = get_num_bfs_paths(self.graph, 'A', 'C', 'distance',
'more_than', 10)
self.assertEqual(number_of_trips, 3)
def test_equal_to_distance(self):
number_of_trips = get_num_bfs_paths(self.graph, 'B', 'B', 'distance',
'equal_to', 9)
self.assertEqual(number_of_trips, 1)
def test_equal_to_stops(self):
number_of_trips = get_num_bfs_paths(self.graph, 'B', 'B', 'towns',
'equal_to', 3)
self.assertEqual(number_of_trips, 1)
def test_less_than_distance(self):
number_of_trips = get_num_bfs_paths(self.graph, 'A', 'B', 'distance',
'less_than', 15)
self.assertEqual(number_of_trips, 3)
def test_more_than_stops(self):
number_of_trips = get_num_bfs_paths(self.graph, 'A', 'C', 'towns',
'more_than', 4)
self.assertEqual(number_of_trips, 0)
def test_bad_value_neg_int(self):
with self.assertRaises(ValueError) as context:
get_num_bfs_paths(self.graph, 'A', 'C', 'towns', 'equal_to', -3)
self.assertTrue('must be an integer' in str(context.exception))
def test_bad_operator(self):
with self.assertRaises(ValueError) as context:
get_num_bfs_paths(self.graph, 'A', 'C', 'towns', 'smells_like', 3)
self.assertTrue('accepted operators are' in str(context.exception))
def test_bad_filter_type(self):
with self.assertRaises(ValueError) as context:
get_num_bfs_paths(self.graph, 'A', 'C', 'grass', 'less_than', 3)
self.assertTrue('accepted filter types are' in str(context.exception))
def test_dest_vertex_doesnt_exist_in_graph(self):
with self.assertRaises(errors.DoesNotExistError) as context:
get_num_bfs_paths(self.graph, 'FEAR', 'C', 'towns', 'less_than', 3)
self.assertTrue('does not exist as a connected vertex in the graph' in
str(context.exception))