class TestPatient(TestCase):
""" Unit tests for the Patient class """
def setUp(self):
""" creates a test fixture before each test method has run """
self.patient = Patient(1, "John Doe", "A", 1, 3, False, False, True,
True, "Jane", "2021-03-18", "2021-03-20",
"comment", "twin")
self.long_str = ""
for i in range(260):
self.long_str += "a"
def test_getters(self):
""" test getters """
self.assertEqual(self.patient.get_id(), 1)
self.assertEqual(self.patient.get_name(), "John Doe")
self.assertEqual(self.patient.get_last_name(), "Doe")
self.assertEqual(self.patient.get_clinical_area(), "A")
self.assertEqual(self.patient.get_bed_num(), 1)
self.assertEqual(self.patient.get_acuity(), 3)
self.assertEqual(self.patient.get_transfer(), False)
self.assertEqual(self.patient.get_a_trained(), False)
self.assertEqual(self.patient.get_one_to_one(), True)
self.assertEqual(self.patient.get_picc(), True)
self.assertEqual(self.patient.get_previous_nurses(), "Jane")
self.patient.set_previous_nurses("Amy")
self.assertEqual(self.patient.get_previous_nurses(), "Amy")
self.assertEqual(self.patient.get_admission_date(), "2021-03-18")
self.assertEqual(self.patient.get_discharge_date(), "2021-03-20")
self.assertEqual(self.patient.get_comment(), "comment")
self.assertEqual(self.patient.get_twin(), "twin")
self.patient.set_assigned(True)
self.assertEqual(self.patient.get_assigned(), True)
def test_invalid_id(self):
"""id"""
with self.assertRaises(ValueError):
Patient("1", "John Doe", "A", 1, 3, False, False, True, True,
"Jane", "2021-03-18", "2021-03-20", "comment", "twin")
with self.assertRaises(ValueError):
Patient(-1, "John Doe", "A", 1, 3, False, False, True, True,
"Jane", "2021-03-18", "2021-03-20", "comment", "twin")
with self.assertRaises(ValueError):
Patient(None, "John Doe", "A", 1, 3, False, False, True, True,
"Jane", "2021-03-18", "2021-03-20", "comment", "twin")
def test_invalid_name(self):
with self.assertRaises(ValueError):
Patient(1, 1, "A", 1, 3, False, False, True, True, "Jane",
"2021-03-18", "2021-03-20", "comment", "twin")
with self.assertRaises(ValueError):
Patient(1, "", "A", 1, 3, False, False, True, True, "Jane",
"2021-03-18", "2021-03-20", "comment", "twin")
with self.assertRaises(ValueError):
Patient(1, self.long_str, "A", 1, 3, False, False, True, True,
"Jane", "2021-03-18", "2021-03-20", "comment", "twin")
def test_to_dict(self):
"""to_dict"""
self.patient.assigned = True
self.assertEqual(
self.patient.to_dict(), {
"id": 1,
"first_name": "John",
"last_name": "Doe",
"clinical_area": "A",
"bed_num": 1,
"acuity": 3,
"a_trained": False,
"transfer": False,
"picc": True,
"one_to_one": True,
"previous_nurses": "Jane",
"admission_date": "2021-03-18",
"discharge_date": "2021-03-20",
"comments": "comment",
"twin": "twin"
})
class TestDepartment(TestCase):
"""This is a test class to test the Department class"""
def setUp(self) -> None:
"""Initialize setup object"""
self.department = Department("Emergency")
self.department1 = Department("Surgery")
self.read_mock = Mock()
self.department._write_to_file = self.read_mock
self.doctor1 = Doctor("George", "Bush", "1982-2-28",
"97334 Oak Bridge , Vancouver, Vancouver, BC", 2,
False, 125, 190000)
self.patient1 = Patient("Jose", "McDonald", "1970-12-12",
"3432 Newtons, Richmond, BC", 1, False, 590)
self.doctor2 = Doctor("Johnny", "Kenedy", "1984-1-30",
"1444 Oakway, North Vancouver, Vancouver, BC", 1,
False, 123, 150000)
self.patient2 = Patient("Bill", "Stark", "1970-12-12",
"1111 Columbia, New Westminster, BC", 2, False,
589, 10000)
self.patient3 = Patient("Tony", "Stark", "1960-9-2",
"1111 Columbia, New Westminster, BC", 12,
False, 589)
def test_valid_constructor(self):
"""Test an object is created correctly"""
self.assertIsNotNone(self.department)
self.assertIsInstance(self.department, Department)
def test_invalid_constructor(self):
"""Test an object with invalid parameters"""
with self.assertRaises(TypeError):
department_1 = Department(1)
def test_add_person(self):
"""Test to add an object (person) to the department"""
with self.assertRaises(ValueError):
self.department.add_person(self.patient1)
self.assertTrue(self.read_mock.called)
# This is to test how many patient objects are added to the self.department
test_case = self.department.get_statistics()
self.assertEqual(test_case.get_not_released_patient_num(), 1)
def test_remove_person(self):
"""Test to remove an object (person) out of the department"""
test_id = self.patient1.get_id()
self.department.remove_person_by_id(test_id)
self.assertTrue(self.read_mock.called)
# This is to test how many patient objects left in the self.department
test_case = self.department.get_statistics()
self.assertEqual(test_case.get_not_released_patient_num(), 0)
# This is to test exception value error of remove method
with self.assertRaises(ValueError):
self.department.remove_person_by_id(100000)
def test_check_existing_person(self):
"""Test to check if an object (person) is in the department"""
test_id = self.patient3.get_id()
self.assertEqual(self.department.person_exist(test_id), False)
test_id_1 = self.doctor2.get_id()
self.assertEqual(self.department.person_exist(test_id_1), True)
def test_get_person_by_type(self):
"""Test to get all people with a given type"""
obj = self.department.get_person_by_type("Teacher")
self.assertEqual(len(obj), 0)
self.department.get_all_current_people()
obj1 = self.department.get_person_by_type("Patient")
self.assertEqual(len(obj1), 2)
def test_get_all_people(self):
"""Test to get all people in a department"""
self.department.get_all_current_people()
def test_get_existing_person_by_id(self):
"""Test to get an object (person) from the department"""
test_id = self.patient1.get_id()
obj1 = self.department.get_person_by_id(test_id)
D1 = obj1.to_dict()
self.assertEqual(D1, self.patient1.to_dict())
obj2 = self.department.get_person_by_id(10000)
self.assertEqual(obj2, None)
def test_get_department_name(self):
"""Test to get the name of the department"""
self.assertEqual(self.department.get_name(), "Emergency")
def test_get_get_statistics(self):
"""Test to get statistics of the department"""
test_case = self.department.get_statistics()
self.assertEqual(test_case.get_not_released_patient_num(), 1)
self.assertEqual(test_case.get_released_patient_num(), 1)
self.assertEqual(test_case.get_total_bill_amount_released_patients(),
10000)
def test_write_data_to_file(self):
"""Test to see if a file is opened and gotten data"""
self.department._write_to_file()
self.assertTrue(self.read_mock.called)
@mock.patch('department.Department._read_from_file')
def test_read_data_to_file(self, mock_read_func):
"""Test to see if a file is opened and data is written on"""
self.department._read_from_file()
self.assertTrue(mock_read_func.called)
def test_get_created_entities(self):
"""Test to check a valid object is created correctly"""
person1 = [{
'first_name': 'George',
'last_name': 'Bush',
'date_of_birth': "1970-12-12 00:00:00",
'address': '97334 Oak Bridge , Vancouver, Vancouver, BC',
'id': 2,
'is_released': False,
'office_num': 125,
'income': 190000
}]
person2 = [{
'first_name': 'Jose',
'last_name': 'McDonald',
'date_of_birth': "1970-12-12 00:00:00",
'address': '3432 Newtons, Richmond, BC',
'is_released': False,
'id': 1,
'room_num': 589,
'bill': 0.0
}]
self.department1.create_entities(person2, "Patient")
self.department1.create_entities(person1, "Doctor")
obj1 = self.department1.get_person_by_id(1)
self.assertIsNotNone(obj1)
self.assertIsInstance(obj1, (Doctor, Patient))
obj2 = self.department1.get_person_by_id(2)
self.assertIsNotNone(obj2)
self.assertIsInstance(obj2, (Doctor, Patient))
def test_update_person(self):
"""Test update information of a person"""
with self.assertRaises(ValueError):
self.department.update_person(4, "Uy", "Tran", 600, 10000)
self.department.update_person(1, "Uy", "Tran", 600, 10000)
self.assertTrue(self.read_mock.called)
def test_to_dict(self):
"""Test to check to_dict() return an expected value"""
self.D1 = self.department1.to_dict()
self.D2 = {
'name':
'Surgery',
'Patient': [{
'first_name': 'Jose',
'last_name': 'McDonald',
'date_of_birth': datetime(1970, 12, 12, 0, 0),
'address': '3432 Newtons, Richmond, BC',
'is_released': False,
'id': 1,
'room_num': 590,
'bill': 0
}, {
'first_name': 'Bill',
'last_name': 'Stark',
'date_of_birth': datetime(1960, 9, 2, 0, 0),
'address': '1111 Columbia, New Westminster, BC',
'is_released': True,
'id': 2,
'room_num': 589,
'bill': 10000
}],
'Doctor': [{
'first_name': 'George',
'last_name': 'Bush',
'date_of_birth': datetime(1982, 2, 28, 0, 0),
'address': '97334 Oak Bridge , Vancouver, Vancouver, BC',
'id': 2,
'is_released': False,
'office_num': 125,
'income': 190000
}]
}
self.assertDictEqual(self.D1, self.D2)
