def test_div(self):
agg = Aggregator(self.comp_parent, [self.comp_1, self.comp_3], Operator.DIV)
agg.aggregate()
self.assertEqual(self.comp_parent.data.equals(pd.Series([0.5, 1, 1.5, 2, 2.5], index=self.idx_1)), True)
def test_sub(self):
agg = Aggregator(self.comp_parent, [self.comp_2, self.comp_3], Operator.SUB)
agg.aggregate()
self.assertEqual(self.comp_parent.data.equals(pd.Series([3, 2, 1, 0, -1], index=self.idx_1)), True)
def test_mul(self):
agg = Aggregator(self.comp_parent, [self.comp_1, self.comp_2, self.comp_3], Operator.MULT)
agg.aggregate()
self.assertEqual(self.comp_parent.data.equals(pd.Series([10, 16, 18, 16, 10], index=self.idx_1)), True)
def test_add(self):
print("Operator est de type: ", type(Operator))
agg = Aggregator(self.comp_parent, [self.comp_1, self.comp_2, self.comp_3], Operator.ADD)
agg.aggregate()
self.assertEqual(self.comp_parent.data.equals(pd.Series([8, 8, 8, 8, 8], index=self.idx_1)), True)