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)