formatted into test cases using unittest