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))