def test_parse_tokens(self):
example_prepped = pp.remove_whitespace(pp.tokenize_and_or(
pp.strip_irrelevant(pp.remove_tags(self.longer_example))))
parsed = pp.parse_tokens(example_prepped)
correct = {'type': 'and', 'courses': [
{'type' : 'or', 'courses' : [u'CS 3510', u'CS 3511']},
u'MATH 3012',
{'type' : 'or', 'courses': [
u'MATH 3215',
u'MATH 3770',
u'MATH 3670',
u'CEE 3770',
u'ISYE 3770',
{'type': 'and', 'courses': [u'ISYE 2027', u'ISYE 2028']}
]}
]}
self.assertEqual(parsed, correct)
def test_sat_act(self):
example_prepped = pp.remove_whitespace(pp.tokenize_and_or(
pp.strip_irrelevant(pp.remove_tags(self.sat_example))))
parsed = pp.parse_tokens(example_prepped)
correct = {'courses': [u'MATH 1113', u'SAT Mathematics 550', u'ACT Math 550'], 'type': 'or'}
self.assertEqual(parsed, correct)
