def test_targets_multi(self):
comment = "blablablabla multi blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'], ['All'])
def test_element_power_medals(self):
comment = "Blablablablabla power blablablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['elements'], ['Power'])
def test_targets_random(self):
comment = "blablablabla random blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'], ['Random'])
def test_element_speed(self):
comment = "Blablablablabla speed blablablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['elements'], ['Speed'])
def test_targets_single(self):
comment = "blablablabla single blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'], ['Single'])
def test_element_magic(self):
comment = "Blablablablabla magic blablablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['elements'], ['Magic'])
def test_direction_upright(self):
comment = "blabla upright blabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['direction'], ['Upright'])
def test_targets_all_doesnt_triger_anything(self):
comment = "blablablabla all blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'], [])
def test_element_psm(self):
comment = "Blablablablabla PSM blablablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['elements'],
['Speed', 'Power', 'Magic'])
def test_element_several_elements(self):
comment = "Blablablablabla power speed blablablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['elements'],
['Power', 'Speed'])
def test_direction_is_case_insensitive(self):
comment = "blabla UPRIGHT REVERSED blabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['direction'],
['Upright', 'Reversed'])
def test_direction_both_directions(self):
comment = "blabla upright reversed blabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['direction'],
['Upright', 'Reversed'])
def test_direction_reversed(self):
comment = "blabla reversed blabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['direction'],
['Reversed'])
def test_targets_is_case_insensitive(self):
comment = "blablablabla SINGLE AOE blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'],
['Single', 'All'])
def test_targets_several(self):
comment = "blablablabla single aoe random blablabla"
extractor = RequirementExtractor(comment)
extractor.extract_requirements()
self.assertCountEqual(extractor.requirements['targets'],
['Single', 'All', 'Random'])
def test_element_is_case_insensitive(self):
comment1 = "Blablablablabla POWER blablablabla"
comment2 = "Blablab SPEED lablabla MAGIC blablablabla"
comment3 = "Blablablablabla PSM blablablabla"
extractor1 = RequirementExtractor(comment1)
extractor2 = RequirementExtractor(comment2)
extractor3 = RequirementExtractor(comment3)
extractor1.extract_requirements()
extractor2.extract_requirements()
extractor3.extract_requirements()
self.assertCountEqual(extractor1.requirements['elements'], ['Power'])
self.assertCountEqual(extractor2.requirements['elements'],
['Magic', 'Speed'])
self.assertCountEqual(extractor3.requirements['elements'],
['Power', 'Magic', 'Speed'])