def test_cc_itemfinder_valid(self): """cc_itemfinder: yield each expected block.""" # pprint(list(cc_itemfinder(labeloff(cc_lines)))) input_with_license = labeloff(cc_lines) self.assertEqual(len(list(cc_itemfinder(input_with_license))), 9) input_without_license = labeloff(cc_lines[:-4]) self.assertEqual(len(list(cc_itemfinder(input_without_license))), 8)
def test_cc_itemfinder_raise(self): """cc_itemfinder: raise RecordError if license block bad.""" input_with_license_lacking_bottom = labeloff(cc_lines[:-1]) self.assertRaises(FieldError, cc_itemfinder, input_with_license_lacking_bottom)
def test_labeloff(self): """labeloff: should return expected lines""" tests = ["AA aa.", "BB bb.", "CC C cc.", "DD dd.", "EE", ""] # expects = [line[5:] for line in tests] expects = ["aa.", " bb.", " cc.", ".", "", ""] self.assertEqual(labeloff(tests), expects)