def test_MP_string(self):
np.random.seed(100)
d = discrete.maxProb(["A", "B", "C"])
result = d([0.2, 0.3, 0.5], trial_responses=["A", "B"])
correct_result = ('B', collections.OrderedDict([('A', 0.2), ('B', 0.3), ('C', 0.5)]))
assert result == correct_result
def test_MP_err_2(self):
np.random.seed(100)
d = discrete.maxProb(task_responses=[1, 2, 3])
result = d([0.6, 0.3, 0.5], trial_responses=[1, 1])
correct_result = (1, collections.OrderedDict([(1, 0.6), (2, 0.3), (3, 0.5)]))
assert result == correct_result
def test_MP_normal_2(self):
np.random.seed(101)
d = discrete.maxProb(task_responses=[1, 2, 3])
result = d([0.5, 0.3, 0.5])
correct_result = (3, collections.OrderedDict([(1, 0.5), (2, 0.3), (3, 0.5)]))
assert result == correct_result