class TestHeroSetFeatureExtractor(unittest2.TestCase):
def setUp(self):
self.hsfe = HeroSetFeatureExtractor()
self.hfe = HeroFeatureExtractor()
def test_output(self):
output = dict()
heroes = ["Abaddon", "Arc Warden", "Death Prophet", "Huskar", "Juggernaut"]
for name, value in zip(self.hsfe.extractFeatureNames(), self.hsfe.extract(heroes)):
output[name] = value
self.assertEqual(len(self.hsfe.extractFeatureNames()), len(self.hsfe.extract(heroes)))
self.assertEqual(output["sum_ATT"], 4)
self.assertEqual(output["max_ATT"], 2)
self.assertEqual(output["min_ATT"], 0)
self.assertEqual(output["median_ATT"], 1)
self.assertEqual(output["product_ATT"], 0)
self.assertEqual(output["sum_FARM PRIORITY"], 0.6105)
self.assertEqual(output["max_FARM PRIORITY"], 0.162)
self.assertEqual(output["min_FARM PRIORITY"], 0.088)
self.assertEqual(output["median_FARM PRIORITY"], 0.1301)
self.assertEqual(output["product_FARM PRIORITY"], 2.3999519522879996e-05)
def setUp(self):
self.hsfe = HeroSetFeatureExtractor()
self.hfe = HeroFeatureExtractor()
