File size: 2,265 Bytes
952e1e7 7027070 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | # SPDX-License-Identifier: MIT
import math
import unittest
from probabilities import mix
class ProbabilityContract(unittest.TestCase):
def setUp(self):
self.r = {'questions': {'q': {'type': 'choice', 'criteria': {'a': 'A', 'b': 'B', 'c': 'C'}}}}
self.g = {'answers': {'q': {'type': 'choice', 'probabilities': {'a': .6, 'b': .3, 'c': .1}}}}
self.q = {'answers': {'q': {'type': 'choice', 'probabilities': {'a': .2, 'b': .3, 'c': .5}}}}
def test_equal_probability_average(self):
answer = mix(self.r, self.g, self.q)['answers']['q']
self.assertEqual(set(answer['probabilities']), {'a','b','c'})
for key, expected in {'a': .4,'b': .3,'c': .3}.items():
self.assertAlmostEqual(answer['probabilities'][key], expected)
self.assertEqual(answer['choice'], 'a')
self.assertAlmostEqual(math.fsum(answer['probabilities'].values()), 1, places=15)
def test_missing_option_rejected(self):
del self.q['answers']['q']['probabilities']['c']
with self.assertRaises(ValueError): mix(self.r,self.g,self.q)
def test_invalid_native_probabilities_rejected(self):
for value in (float('nan'),float('inf'),-1,2):
self.q['answers']['q']['probabilities']['c']=value
with self.assertRaises(ValueError): mix(self.r,self.g,self.q)
def test_unnormalized_rejected(self):
self.q['answers']['q']['probabilities']={'a':.1,'b':.1,'c':.1}
with self.assertRaises(ValueError): mix(self.r,self.g,self.q)
def test_question_mismatch_rejected(self):
self.q['answers']={}
with self.assertRaises(ValueError): mix(self.r,self.g,self.q)
def test_noul_and_score(self):
r={'questions':{'n':{'type':'noul'},'s':{'type':'score','criteria':['bad','good']}}}
g={'answers':{'n':{'type':'noul','noul':.2},'s':{'type':'score','probabilities':{'0':.3,'1':.7}}}}
q={'answers':{'n':{'type':'noul','noul':.8},'s':{'type':'score','probabilities':{'0':.7,'1':.3}}}}
result=mix(r,g,q)['answers']
self.assertEqual(result['n']['noul'],.5)
self.assertEqual(result['s']['score'],.5)
self.assertEqual(result['s']['probabilities'],{'0':.5,'1':.5})
if __name__=='__main__': unittest.main()
|