def test_normalizer(self): eq_result = ParseResult('Equation', None, 90) pc_result = ParseResult('Postal Code', None, 80) results = EquationWithPhonePostalCode.normalize([eq_result, pc_result]) count = 0 for res in results: if res.type == 'Equation': count += 1 self.assertEqual(res.confidence, 75) elif res.type == "Postal Code": count += 1 self.assertEqual(res.confidence, 80) self.assertEqual(count, len(results))
def test_test(self): self.assertFalse( EquationWithPhonePostalCode.test(['Equation', 'Boolean'], [])) self.assertTrue( EquationWithPhonePostalCode.test(['Equation', 'Postal Code'], []))