def test_extract_section_features(self): normalizer = Normalizer() self.assertEqual(normalizer.extract_section_features('136'), generate_feature(d='136')) self.assertEqual(normalizer.extract_section_features('Reserve 40 '), generate_feature(pp="reserve", d='40')) self.assertEqual(normalizer.extract_section_features('Top Deck 6 '), generate_feature(pp="top deck", d='6')) self.assertEqual(normalizer.extract_section_features('31RS '), generate_feature(d='31', s='rs')) self.assertEqual(normalizer.extract_section_features('Left Field Pavilion 311'), generate_feature(pp='left field pavilion', d='311')) self.assertEqual(normalizer.extract_section_features('Infield Reserve IFR7 '), generate_feature(pp='infield reserve', p='ifr', d='7')) self.assertEqual(normalizer.extract_section_features('311PL'), generate_feature(d='311', s='pl')) self.assertEqual(normalizer.extract_section_features('F9'), generate_feature(p='f', d='9')) self.assertEqual(normalizer.extract_section_features('L36'), generate_feature(p='l', d='36'))