partially fixed test_sap_model_prep

This commit is contained in:
Khalim Conn-Kowlessar 2024-01-06 18:22:37 +00:00
parent 756dca12ba
commit 6335be531f

View file

@ -2,7 +2,6 @@ from backend.Property import Property
from etl.epc.DataProcessor import DataProcessor
from backend.app.plan.utils import create_recommendation_scoring_data, get_cleaned
from etl.epc.settings import COLUMNS_TO_MERGE_ON
from epc_api.client import EpcClient
import pandas as pd
import pytest
import msgpack
@ -288,8 +287,7 @@ class TestSapModelPrep:
home = Property(
id=0,
postcode=starting_epc["postcode"],
address1=starting_epc["address1"],
epc_client=EpcClient(auth_token="notoken"),
address=starting_epc["address1"],
data=starting_epc
)
home.get_components(cleaned)
@ -508,8 +506,7 @@ class TestSapModelPrep:
home2 = Property(
id=0,
postcode=starting_epc2["postcode"],
address1=starting_epc2["address1"],
epc_client=EpcClient(auth_token="notoken"),
address=starting_epc2["address1"],
data=starting_epc2
)
home2.get_components(cleaned)
@ -728,8 +725,7 @@ class TestSapModelPrep:
home3 = Property(
id=0,
postcode=starting_epc3["postcode"],
address1=starting_epc3["address1"],
epc_client=EpcClient(auth_token="notoken"),
address=starting_epc3["address1"],
data=starting_epc3
)
home3.get_components(cleaned)
@ -937,8 +933,7 @@ class TestSapModelPrep:
home4 = Property(
id=0,
postcode=starting_epc4["postcode"],
address1=starting_epc4["address1"],
epc_client=EpcClient(auth_token="notoken"),
address=starting_epc4["address1"],
data=starting_epc4
)
home4.get_components(cleaned)