def test_capacity_invalid_type(self):
""" Test that invalid type raises exception """
with self.assertRaises(Exception):
CourseVal.capacity("1234")
with self.assertRaises(Exception):
CourseVal.capacity(-1234)
def test_capacity(self):
""" Test that valid args does not raise exception """
try:
CourseVal.capacity(1234)
except Exception:
self.fail()