def test_get_assignments(self): factor = Factor(self.random_variables) assignments = factor.get_random_variables_assignments() self.assertEqual( {(1, 3, 2), (1, 4, 2), (1, 5, 2), (2, 3, 2), (2, 4, 2), (2, 5, 2)}, set(assignments))