Skip to content

Commit 1a49ad7

Browse files
Added Tests for employee schema
Co-Authored-By: Carrington <215159144+carrington-cs@users.noreply.github.com>
1 parent 734a8a2 commit 1a49ad7

File tree

3 files changed

+43
-0
lines changed

3 files changed

+43
-0
lines changed

examples/with-unittest/library_management_system/main.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,19 @@ def __init__(self, title, author, year_published, isbn, typr_ = "book"):
5353

5454
def __str__(self):
5555
return f"ReadingMaterialUnit({self.title=}, {self.author=}, {self.isbn=})"
56+
57+
class Employee:
58+
def __init__(self, name: str, profession: str, age: str, race: str = "african"):
59+
self.name = name
60+
self.age = age
61+
self.profession = profession
62+
self.race = race
5663

64+
def __str__(self):
65+
return f"Employee({self.name=}, {self.profession=}, {self.age=})"
66+
5767
reading_material_unit = ReadingMaterialUnit("New Maths", "author", "isbn", "year_published")
5868
reading_material_unit_2 = reading_material_unit
69+
5970
print(hex(id(reading_material_unit)))
6071
print(hex(id(reading_material_unit_2)))
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
import unittest
2+
3+
from main import Employee
4+
5+
class TestEmployee(unittest.TestCase):
6+
def setUp(self):
7+
"""
8+
Docstring for setUp
9+
10+
:param employee: Employee of the Library
11+
"""
12+
self.employee = Employee(name="name", age=19, profession="cleaner")
13+
14+
15+
def tearDown(self):
16+
self.employee = None
17+
18+
def test_employee_created(self):
19+
self.assertTrue(type(self.employee.name) == str)
20+
self.assertTrue(type(self.employee.profession) == str)
21+
self.assertTrue(type(self.employee.age) == int)
22+
self.assertTrue(type(self.employee.race) == str)
23+
24+
def test_employee_has_not_retired(self):
25+
self.assertGreaterEqual((self.employee.age), 18)
26+
27+
def test_employee_has_race(self):
28+
self.assertEqual(type(self.employee.race), str)
29+
30+
def test_employee_string_representation(self):
31+
self.assertIn(self.employee.name, str(self.employee))
32+
self.assertIn(str(self.employee.age), str(self.employee))

examples/with-unittest/library_management_system/utils.py

Whitespace-only changes.

0 commit comments

Comments
 (0)