def test_VP_reduced_str(self):
correct_result = (np.array([0.1, 0.7]), np.array(['B', 'C']))
result = discrete._validProbabilities([0.2, 0.1, 0.7], ["A", "B", "C"], ["B", "C"])
assert (result[0] == correct_result[0]).all()
assert (result[1] == correct_result[1]).all()
def test_VP_err_2(self):
correct_result = (np.array([0.2]), np.array(['A']))
result = discrete._validProbabilities([0.2, 0.1, 0.7], ["A", "B", "C"], ["A", "A"])
assert (result[0] == correct_result[0]).all()
assert (result[1] == correct_result[1]).all()
def test_VP_reduced_int(self):
correct_result = (np.array([0.1, 0.7]), np.array([2, 3]))
result = discrete._validProbabilities([0.2, 0.1, 0.7], [1, 2, 3], [2, 3])
assert (result[0] == correct_result[0]).all()
assert (result[1] == correct_result[1]).all()