def test_strip_irrelevant_handles_grade(self):
strings = []
strings.append(pp.strip_irrelevant("CS 1332 Minimum Grade of C"))
strings.append(pp.strip_irrelevant("CS 2340 Minimum Grade of D"))
strings.append(pp.strip_irrelevant("( CS 2340 Minimum Grade of F)"))
self.assertEqual(strings[0], "CS 1332")
self.assertEqual(strings[1], "CS 2340")
self.assertEqual(strings[2], "( CS 2340)")
def test_strip_irrelevant_handles_level(self):
strings = []
strings.append(pp.strip_irrelevant("Undergraduate Semester level CS 1332 or Graduate Semester level CS 2110"))
strings.append(pp.strip_irrelevant("Undergraduate Semester level CS 2340"))
strings.append(pp.strip_irrelevant("( Graduate Semester level CS 2340)"))
self.assertEqual(strings[0], " CS 1332 or CS 2110")
self.assertEqual(strings[1], " CS 2340")
self.assertEqual(strings[2], "( CS 2340)")
def test_strip_irrelevant_handles_both(self):
strings = []
strings.append(pp.strip_irrelevant("(Undergraduate Semester level CS 3510 Minimum Grade"
" of C or Undergraduate Semester level CS 3511 Minimum Grade of C)"))
strings.append(pp.strip_irrelevant("and Undergraduate Semester level MATH 3012 Minimum Grade of D and"))
strings.append(pp.strip_irrelevant(self.long_example))
# a rediculous real world example
self.assertEqual(strings[0], "( CS 3510 or CS 3511)")
self.assertEqual(strings[1], "and MATH 3012 and")
self.assertEqual(strings[2], ' MATH 3215 or MATH 3770 or MATH 3670 or CEE 3770 or ISYE 3770 or ( ISYE 2027 and ISYE 2028) )')
def test_full_series(self):
longest_example_list = pp.remove_whitespace(pp.tokenize_and_or(
pp.strip_irrelevant(pp.remove_tags(self.longer_example))))
self.assertEqual(longest_example_list, ['(', 'CS 3510', 'or', 'CS 3511', ')', 'and',
'MATH 3012', 'and', '(', 'MATH 3215', 'or', 'MATH 3770', 'or', 'MATH 3670',
'or', 'CEE 3770', 'or', 'ISYE 3770', 'or', '(', 'ISYE 2027', 'and', 'ISYE 2028',
')', ')'])
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)