From 3a49aecbf60572ddeb1294b17a36c7ab6f959183 Mon Sep 17 00:00:00 2001 From: HighestAuto Date: Mon, 19 Aug 2024 16:49:42 +0100 Subject: [PATCH] checking if outputs are lists --- tests/test_transform.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/test_transform.py b/tests/test_transform.py index 6a4361d..bb092b0 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -11,6 +11,7 @@ class TestTransform: def test_generation_transform_general(self, mock_gen_df): trans_df = transform.generation_transform(mock_gen_df) + assert isinstance(trans_df, list) for row in trans_df: assert len(row) == 6 assert row[1] in row[0] @@ -34,6 +35,7 @@ def test_generation_transform_valid_dates(self, mock_gen_df): def test_demand_transform_general(self, mock_demand_df): trans_df = transform.demand_transform(mock_demand_df) + assert isinstance(trans_df, list) for row in trans_df: assert len(row) == 2 time = datetime.datetime.strptime(row[0], "%Y-%m-%dT%H:%M:%SZ") @@ -42,6 +44,7 @@ def test_demand_transform_general(self, mock_demand_df): def test_cost_transform_general(self, mock_cost_df): trans_df = transform.cost_transform(mock_cost_df) + assert isinstance(trans_df, list) for row in trans_df: assert len(row) == 4 date = datetime.datetime.strptime(row[0], "%Y-%m-%d") @@ -51,8 +54,8 @@ def test_cost_transform_general(self, mock_cost_df): assert isinstance(row[3], int) == True def test_carbon_transform_general(self, mock_carbon_df): - trans_df = transform.carbon_transform(mock_carbon_df) + assert isinstance(trans_df, list) for row in trans_df: assert len(row) == 3 time = datetime.datetime.strptime(row[0], "%Y-%m-%dT%H:%MZ") @@ -61,7 +64,7 @@ def test_carbon_transform_general(self, mock_carbon_df): assert row[2] in ["very low", "low", "moderate", "high", "very high"] - def test_carbon_transform_general(self, mock_carbon_df): + def test_carbon_transform_adjust_carbon_levels(self, mock_carbon_df): assert mock_carbon_df['carbon level'][2] == 'low' assert mock_carbon_df['carbon level'][2] == 'low' assert mock_carbon_df['carbon level'][2] == 'low' @@ -79,6 +82,7 @@ def test_get_difference_of_dates(self, mock_demand_df, mock_gen_df): gen_tuple = transform.generation_transform(mock_gen_df) data = {'generation': gen_tuple, 'demand': demand_tuple} data = transform.difference_of_dates(data) + assert isinstance(data, dict) times = [] for values in data['demand']: times.append(values[0])