def main():
staff1 = Staff(50)
staff2 = Staff(75)
staff3 = Staff(50)
staff4 = Staff(-10)
#Testing Staff.getPower()
print()
print("Testing Staff.getPower().")
print("Test Staff.getPower(), expected 50 was", staff1.getPower())
print("Test Staff.getPower(), expected 75 was", staff2.getPower())
print("Test Staff.getPower(), expected 50 was", staff3.getPower())
print("Test Staff.getPower(), expected 25 was", staff4.getPower())
print()
#Testing Staff == and !=
print("Testing Staff == and !=.")
print("Test Staff ==, expected True was", staff1 == staff3)
print("Test Staff ==, expected False was", staff2 == staff3)
print("Test Staff !=, expected True was", staff1 != staff4)
print("Test Staff !=, expected False was", staff1 != staff3)
print()
#Testing Staff string representation
print("Testing Staff string representation.")
print('Expected "Staff of power 50." was "%s"' % str(staff1))
print('Expected "Staff of power 75." was "%s"' % str(staff2))
print()
wizard1 = Wizard("Hermione", staff1)
wizard2 = Wizard("Ron", staff2)
#Testing Wizard.getName()
print("Testing Wizard.getName()")
print("Expected Hermione was", wizard1.getName())
print("Expected Ron was", wizard2.getName())
print()
#Testing Wizard.getStaff()
print("Testing Wizard.getStaff()")
print('Expected "Staff of power 50." was "%s"' % str(wizard1.getStaff()))
print('Expected "Staff of power 75." was "%s"' % str(wizard2.getStaff()))
print()
#Testing Wizard.getPower()
print("Testing Wizard.getPower().")
print("Test Wizard.getPower(), expected 50 was", wizard1.getPower())
print("Test Wizard.getPower(), expected 75 was", wizard2.getPower())
print()
#Testing Wizard string representation.
print("Testing Wizard string representation.")
print('Expected "Wizard Hermione with staff of power 50."\n was "%s"' %
str(wizard1))
print('Expected "Wizard Ron with staff of power 75."\n was "%s"' %
str(wizard2))
print()
#some battles
#Expected: Hermione defeats Draco
battle(Wizard("Hermione", Staff(1000)), Wizard("Draco", Staff(250)))
print()
#Expected: Ron defeats Draco
battle(Wizard("Draco", Staff(500)), Wizard("Ron", Staff(1500)))
print()
#Expected: It's a draw
battle(Wizard("Hermione", Staff(1)), Wizard("Harry", Staff(1)))
print()
#Expected: Unknown!
battle(Wizard("Harry", Staff(100)), Wizard("Hermione", Staff(100)))
print()
