def test_bool_to_num(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.bool_to_num(True), 1, 'Should be 1')
self.assertEqual(test.bool_to_num(False), 0, 'Should be 0')
with self.assertRaises(Exception):
test.bool_to_num(55)
def test_risk_from_predicted_cases(self):
calcRisk = RiskAssessment(
20, True, True, True, True, True, 2, True, True,
StateMetrics([100, 120], 1, 150, 0.99, [0.15], 150))
self.assertEqual(calcRisk.risk_from_predicted_cases(), 1,
"Should be 1")
calcRisk.stateInfo.percent_change = [0.75]
self.assertEqual(calcRisk.risk_from_predicted_cases(), 0,
"Should be 0")
def test_set_vaccinated(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.vaccinated, True, 'Should be True')
test.set_vaccinated(False)
self.assertEqual(test.vaccinated, False, 'Should be False')
with self.assertRaises(Exception):
test.set_vaccinated(55)
def test_set_level_of_contact(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.level_of_contact, 2, 'Should be 2')
test.set_level_of_contact(1)
self.assertEqual(test.level_of_contact, 1, 'Should be 1')
with self.assertRaises(Exception):
test.set_level_of_contact(True)
def test_set_age(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.age, 20, 'Should be 20')
test.set_age(50)
self.assertEqual(test.age, 50, 'Should be 50')
with self.assertRaises(Exception):
test.set_age('thirty')
def test_persistent_cough(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.persistent_cough, True, 'Should be True')
test.set_presistent_cough(False)
self.assertEqual(test.persistent_cough, False, 'Should be False')
with self.assertRaises(Exception):
test.set_presistent_cough(55)
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')
'''
def test_get_name(self):
obj = StateMetrics([100, 110, 120], 50, 0.99, 0.01)
s = State('Bob Ross', obj)
self.assertEqual(s.get_name(), 'Bob Ross', "Should be 'Bob Ross'")
def test_get_metrics(self):
obj = StateMetrics([100, 110, 120], 50, 0.99, 0.01)
s = State('Bob Ross', obj)
self.assertEqual(s.get_metrics, obj, "Should store object of StateMetrics")
def test_get_percent_change(self):
s = StateMetrics([110, 100, 90], 100, 0.99, 0.5)
self.assertEqual(s.get_percent_change(), 0.5, 'Should be 0.5')
def test_predict_days_out(self):
s = StateMetrics([110, 100, 90], 100, 0.99, 0.5)
self.assertEqual(s.predictions[1], 100, 'Should be 100')
def test_get_predictions(self):
s = StateMetrics(np.array([110, 100, 90]), 100, 0.99, 0.5)
array_test = np.array([110, 100, 90])
self.assertTrue(np.array_equal(s.predictions, array_test),
'Should be True')
def test_get_avg_cases_per_day(self):
s = StateMetrics([110, 100, 90], 100, 0.99, 0.5)
self.assertEqual(s.get_avg_cases_per_day(), 100, 'Should be 100')
def test_get_model_accuracy(self):
s = StateMetrics([110, 100, 90], 100, 0.99, 0.5)
self.assertEqual(s.get_model_accuracy(), 0.99, 'Should be 0.99')