def test_get_recom(self): my_problem = MatchingProblem(users, items, trans) result = my_problem.get_recom(2, 3) expected = 6 self.assertEqual(result, expected)