def createUnit(self, university):
print("\n \n \n")
print("CREATING A NEW UNIT:")
newUnit = Unit()
code = input("New unit code: ")
valid, code = validator.validateUnitCode(code)
if valid:
newUnit.setCode(code)
title = input("New unit title: ")
newUnit.setTitle(title)
university.addUnit(newUnit)
print("\nSUCCESS: UNIT RECORD CREATED: \n")
newUnit.displayDetails()
print("\n \n \n")
else:
print("ERROR: INVALID UNIT CODE FORMAT")
def createUnit(self,university):
print("\n \n \n")
print("CREATING A NEW UNIT:")
newUnit = Unit()
code = input("New unit code: ")
valid, code = validator.validateUnitCode(code)
if valid:
newUnit.setCode(code)
title = input("New unit title: ")
newUnit.setTitle(title)
university.addUnit(newUnit)
print("\nSUCCESS: UNIT RECORD CREATED: \n")
newUnit.displayDetails()
print("\n \n \n")
else:
print("ERROR: INVALID UNIT CODE FORMAT")
def editUnitDetails(self, university):
print("\n \n \n")
print("EDITING UNIT DETAILS:")
code = input("Enter a Unit code: ")
if university.unitExists(code):
university.units.pop(code, None)
newUnit = Unit()
code = input("New unit code: ")
code = validator.validateUnitCode(code)
newUnit.setCode(code)
title = input("New unit title: ")
newUnit.setTitle(title)
university.addUnit(newUnit)
print("\nSUCCESS: UNIT RECORD CREATED")
newUnit.displayDetails()
else:
print("ERROR: Unit code does not exist")
print("\n \n \n")
def editUnitDetails(self,university):
print("\n \n \n")
print("EDITING UNIT DETAILS:")
code = input("Enter a Unit code: ")
if university.unitExists(code):
university.units.pop(code, None)
newUnit = Unit()
code = input("New unit code: ")
code = validator.validateUnitCode(code)
newUnit.setCode(code)
title = input("New unit title: ")
newUnit.setTitle(title)
university.addUnit(newUnit)
print("\nSUCCESS: UNIT RECORD CREATED")
newUnit.displayDetails()
else:
print("ERROR: Unit code does not exist")
print("\n \n \n")
class UnitClassUnitTest(unittest.TestCase):
'''
def setUp(self):
self.unit = Unit()
self.file = open("filedirect", "r")
def tearDown(self):
self.file.close()
'''
def setUp(self):
self.unit = Unit()
def testA(self):
self.unit.setCode('FIT1004')
assert self.unit.code == 'FIT1004', "setCode not functioning correctly"
def testB(self):
self.unit.setTitle('Databases')
assert self.unit.title == 'Databases', "setTitle not functioning correctly"
def testC(self):
self.unit.addCourse('2770')
assert '2770' in self.unit.courses, "course code not being added correctly"
def testD(self):
self.unit.addStudent('10000000')
assert '10000000' in self.unit.students, "student ID not being added correctly"
def testE(self):
self.unit.code = 'FIT1004'
assert self.unit.getCode() == 'FIT1004', "correct unit code isn't being returned"
def testF(self):
self.unit.title = 'Databases'
assert self.unit.getTitle() == 'Databases', "correct title not being returned"
class UnitClassUnitTest(unittest.TestCase):
'''
def setUp(self):
self.unit = Unit()
self.file = open("filedirect", "r")
def tearDown(self):
self.file.close()
'''
def setUp(self):
self.unit = Unit()
def testA(self):
self.unit.setCode('FIT1004')
assert self.unit.code == 'FIT1004', "setCode not functioning correctly"
def testB(self):
self.unit.setTitle('Databases')
assert self.unit.title == 'Databases', "setTitle not functioning correctly"
def testC(self):
self.unit.addCourse('2770')
assert '2770' in self.unit.courses, "course code not being added correctly"
def testD(self):
self.unit.addStudent('10000000')
assert '10000000' in self.unit.students, "student ID not being added correctly"
def testE(self):
self.unit.code = 'FIT1004'
assert self.unit.getCode(
) == 'FIT1004', "correct unit code isn't being returned"
def testF(self):
self.unit.title = 'Databases'
assert self.unit.getTitle(
) == 'Databases', "correct title not being returned"
