def test_09(self):
# Bacon path, large database, does not exist
actor_1 = 43011
actor_2 = 1204555
expected = None
result = lab.get_path(self.db_large, actor_1, actor_2)
self.assertEqual(result, expected)
def test_06(self):
# Actor path, large database, length of 7 (8 actors, 7 movies)
actor_1 = 1345462
actor_2 = 89614
len_expected = 7
first_result = lab.get_path(self.db_large, actor_1, actor_2)
len_first_result = -1 if first_result is None else len(first_result)-1
self.assertTrue(valid_path(self.db_large, first_result))
self.assertEqual(len_first_result, len_expected)
self.assertEqual(first_result[0], actor_1)
self.assertEqual(first_result[-1], actor_2)
second_result = lab.get_path(self.db_large, actor_1, actor_2)
len_second_result = -1 if second_result is None else len(second_result)-1
self.assertTrue(valid_path(self.db_large, second_result))
self.assertEqual(len_second_result, len_expected)
self.assertEqual(second_result[0], actor_1)
self.assertEqual(second_result[-1], actor_2)
def test_08(self):
# Bacon path, large database, length of 7 (8 actors, 7 movies)
actor_1 = 43011
actor_2 = 1379833
len_expected = 7
result = lab.get_path(self.db_large, actor_1, actor_2)
len_result = -1 if result is None else len(result) - 1
self.assertTrue(valid_path(self.db_large, result))
self.assertEqual(len_result, len_expected)
self.assertEqual(result[0], actor_1)
self.assertEqual(result[-1], actor_2)
def test_07(self):
# Actor path, large database, length of 4 (5 actors, 4 movies)
actor_1 = 100414
actor_2 = 57082
len_expected = 4
result = lab.get_path(self.db_large, actor_1, actor_2)
len_result = -1 if result is None else len(result) - 1
self.assertTrue(valid_path(self.db_large, result))
self.assertEqual(len_result, len_expected)
self.assertEqual(result[0], actor_1)
self.assertEqual(result[-1], actor_2)
def test_get_path(self):
# length of 2
actor_id_1 = 46866
actor_id_2 = 2876
len_expected = 2
result = lab.get_path(self.data, actor_id_1, actor_id_2)
len_result = -1 if result is None else len(result) - 1
self.assertTrue(valid_path(self.data, result))
self.assertEqual(len_result, len_expected)
self.assertEqual(result[0], actor_id_1)
self.assertEqual(result[-1], actor_id_2)
def test_path(self):
self.assertEqual(lab.get_path(self.data, 1532, 2876),
[1532, 4724, 2876])
def test_get_path(self):
result = lab.get_path(self.data,2876,46866)
expected = [2876,1640,46866]
self.assertEqual(result,expected)
def test_case_6(self):
"""longer path"""
result = lab.get_path(self.data, 4724, 46866)
expected = [4724, 2876, 1640, 46866]
self.assertEqual(result, expected)
def test_case_5(self):
"""the path is [actor_1,actor_2] because they acted together\n"""
result = lab.get_path(self.data, 1640, 2876)
expected = [1640, 2876]
self.assertEqual(result, expected)
