def test_build_derivatives_nodata(self):
''' Test the derivate objects is added with no_data flag '''
options = {'cd_analysis_unit': 99}
rules = {
"instances": [{"name": "inst_1", 'type': 'from_id', 'named_prop': 'col_2'}]
}
sources = {"dataset": SAMPLE_DATAFRAME.copy()}
(der_data, der_anynodata) = BaseModel.build_derivatives(
rules,
options,
sources,
{}
)
self.assertEqual(
der_data["inst_1"],
None
)
self.assertEqual(der_anynodata, True)
def test_build_derivatives(self):
''' Test if derivate object is added to the collection '''
options = {'cd_analysis_unit': 2}
rules = {
"instances": [{"name": "inst_1", 'type': 'max', 'named_prop': 'col_2'}]
}
sources = {"dataset": SAMPLE_DATAFRAME.copy()}
(der_data, der_anynodata) = BaseModel.build_derivatives(
rules,
options,
sources,
{}
)
self.assertEqual(
der_data["inst_1"].to_dict(),
{"col_1": "d", "col_2": 3, "col_3": 3}
)
self.assertEqual(der_anynodata, False)
