junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove get_cleaned from unit tests

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2023-11-17 14:09:22 +00:00
parent ee0fd1452b
commit 1846b8dc14
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
backend/tests/test_sap_model_prep.py
View file

@ -5,8 +5,9 @@ from etl.epc.settings import COLUMNS_TO_MERGE_ON
from epc_api.client import EpcClient
import pandas as pd
import pytest
import msgpack
from utils.s3 import read_dataframe_from_s3_parquet
from utils.s3 import read_dataframe_from_s3_parquet, read_from_s3
from tqdm import tqdm
@ -113,6 +114,12 @@ class TestSapModelPrep:
@pytest.fixture
def cleaned(self):
cleaned = read_from_s3(
s3_file_name="cleaned_epc_data/cleaned.bson",
bucket_name="retrofit-data-dev"
)
cleaned = msgpack.unpackb(cleaned, raw=False)
return get_cleaned()
def test_fill_cavity_wall(self, cleaned, cleaning_data):