class DeleteBusinessTest(unittest.TestCase): """Illustrate test cases to test expected behavior of delete business functionality. """ def setUp(self): """Instantiate the Business class so that it can be reused by other test cases.""" self.business = Business() def test_business_id_existence(self): """Test if a KeyError will be raised when the business id does not exist.""" self.business.create_business(1, 'Cosmas', 'Cosma Tech', 'Nairobi', 'Technology', 'Masters of ecommerce') self.assertEqual(self.business.delete_business(4), 'Business does not exist') def test_empty_business_id(self): """Test whether no business id is provided.""" self.assertEqual(self.business.delete_business(''), 'Business ID is required!') def test_negative_integer_business_id_raises_ValueError(self): """Test if ValueError is raised when business id a negative number.""" self.assertEqual(self.business.delete_business(-1), 'The business ID must be a positive number') def test_non_integer_business_id_raises_TypeError(self): """Test if TypeError is raised when business id not a number.""" self.assertEqual(self.business.delete_business('1'), 'The business ID must be an number!')
def delete_business(business_id, user_id): """docstring for delete business""" return Business.delete_business(business_id, user_id)