def test_transform(self):
aws_burn_rate = AwsBurnRate.AwsBurnRate(config)
res = aws_burn_rate.transform(data_block)
assert produces.sort() == list(res.keys()).sort()
expected_df = expected_transform_output[produces[0]]
res_df = res[produces[0]]
assert np.isclose(expected_df["BurnRate"], res_df["BurnRate"])
def test_transform(self):
aws_burn_rate = AwsBurnRate.AwsBurnRate(config)
res = aws_burn_rate.transform(data_block)
verify_products(aws_burn_rate, res)
expected_df = expected_transform_output["AWS_Burn_Rate"]
res_df = res["AWS_Burn_Rate"]
assert np.isclose(expected_df["BurnRate"], res_df["BurnRate"])
def test_produces(self):
aws_burn_rate = AwsBurnRate.AwsBurnRate(config)
assert aws_burn_rate.produces() == produces