From f1c59157ed989cc965519e4edec59c4f75100724 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Tue, 12 Sep 2023 14:00:53 +0100 Subject: [PATCH] Fixed lighting unit tests --- model_data/tests/test_lighting_attributes.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/model_data/tests/test_lighting_attributes.py b/model_data/tests/test_lighting_attributes.py index 64044abd..392e10d7 100644 --- a/model_data/tests/test_lighting_attributes.py +++ b/model_data/tests/test_lighting_attributes.py @@ -5,13 +5,11 @@ from model_data.epc_attributes.LightingAttributes import LightingAttributes # An example averages dataset to use in tests. It is a dictionary where the key is a lighting description and the # value is the expected proportion. -averages = pd.DataFrame( - [ - {"lighting-description": "good lighting efficiency", "low-energy-lighting": 0.75}, - {"lighting-description": "excellent lighting efficiency", "low-energy-lighting": 1.0}, - {"lighting-description": "below average lighting efficiency", "low-energy-lighting": 0.25} - ] -) +averages = [ + {"lighting-description": "good lighting efficiency", "low-energy-lighting": 0.75}, + {"lighting-description": "excellent lighting efficiency", "low-energy-lighting": 1.0}, + {"lighting-description": "below average lighting efficiency", "low-energy-lighting": 0.25} +] class TestLightingAttributes: