def test_set_loss_of_smell_and_taste(self):
test = RiskAssessment(
20, True, True, True, True, True, 2, True, True,
StateMetrics([100, 120], 1, 150, 0.99, [0.15], 150))
self.assertEqual(test.loss_of_smell_and_taste, True, 'Should be True')
test.set_loss_of_smell_and_taste(False)
self.assertEqual(test.loss_of_smell_and_taste, False,
'Should be False')
with self.assertRaises(Exception):
test.set_loss_of_smell_and_taste(55)
def test_risk_assessment(self):
test = RiskAssessment(
20, True, True, True, True, True, 2, True, True,
StateMetrics([100, 120], 1, 150, 0.99, [0.15], 150))
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_vaccinated(False)
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_level_of_contact(1)
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_level_of_contact(0)
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_immuno_compromised(False)
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_severe_fatigue(False)
test.set_riskAssessment()
self.assertTrue(test.risk_value > 0.85, 'Should be greater than 0.5')
test.set_skipped_meals(False)
test.set_riskAssessment()
self.assertTrue((test.risk_value < 0.85) and (test.risk_value > 0.5),
'Should be greater than 0.5')
test.set_presistent_cough(False)
test.set_riskAssessment()
self.assertTrue((test.risk_value < 0.85) and (test.risk_value > 0.5),
'Should be greater than 0.5')
test.set_loss_of_smell_and_taste(False)
test.set_riskAssessment()
self.assertTrue(test.risk_value < 0.5, 'Should be greater than 0.5')
test.set_sex(False)
test.set_riskAssessment()
self.assertTrue(test.risk_value < 0.5, 'Should be greater than 0.5')
test.set_age(55)
test.set_riskAssessment()
self.assertTrue(test.risk_value < 0.5, 'Should be greater than 0.5')
'''