tests/test_preprocess.py

# -*- coding: utf-8 -*-


from app import utils
from app import schemas
from app import preprocess
import pytest
import json


encoder = utils.load_model("onehotencoder.joblib")
scaler = utils.load_model("scaler.joblib")


@pytest.fixture
def data_example():
    """ return data example for input """

    with open("tests/input_data.json", "r", encoding="utf-8") as f:
        data_dict = json.load(f)
        
    data = schemas.InputData(**data_dict)
    return data


def test_encode(data_example):
    """ check the response of encode data """
    x_exemple_scaled = preprocess.encode(data_example)