def test_transformer_with_multiple_expression(self):
test_df = pd.DataFrame(
{
'code': [1, 2, 3, 4, 1, 2, 3],
'b': [4, 5, 6, 7, 6, 5, 4],
'c': [9, 8, 7, 6, 5, 4, 3]
},
index=[1, 1, 1, 1, 2, 2, 2],
dtype=float)
expression1 = SecurityMovingMax(20, 'b')
expression2 = SecurityMovingMin(20, 'c')
expression3 = SecurityMovingMax(20, 'b') + SecurityMovingMin(20, 'c')
calculated = transform(test_df,
[expression1, expression2, expression3],
cols=['factor1', 'factor2', 'factor3'],
category_field='code')
expected = [13., 13., 13., 13., 11., 9, 9.]
np.testing.assert_array_almost_equal(calculated['factor3'], expected)
np.testing.assert_array_almost_equal(
calculated['factor1'] + calculated['factor2'],
calculated['factor3'])
def test_transformer_without_category_name(self):
test_df = pd.DataFrame({'b': [4, 5, 6, 7, 6, 5, 4],
'c': [9, 8, 7, 6, 5, 4, 3]},
index=[1, 2, 3, 4, 5, 6, 7],
dtype=float)
expression = SecurityMovingMax(20, 'b') + SecurityMovingMinimum(20, 'c')
calculated = transform(test_df, [expression], cols=['user_factor'])
expected = [13., 13., 13., 13., 12., 11., 10.]
np.testing.assert_array_almost_equal(calculated['user_factor'], expected)