import pandas as pd import pytest from funda_scraper.preprocess import preprocess_data @pytest.fixture def input_data(): data = { "url": ["https://www.funda.nl/koop/utrecht/appartement-00000000-dummy-100/"], "price": ["na"], "address": ["dummy 10"], "descrip": ["dummy"], "listed_since": ["Verkocht"], "zip_code": ["1111 AA"], "size": ["100 m²"], "year": ["2000"], "living_area": ["78 m²"], "perceel_area": ["78 m²"], "kind_of_house": ["Eengezinswoning"], "building_type": ["Bestaande bouw"], "num_of_rooms": ["4 kamers (3 slaapkamers)"], "num_of_bathrooms": ["1 badkamer en 1 apart toilet"], "layout": ["Aantal kamers 4 kamers (3 slaapkamers)"], "energy_label": ["A++++"], "insulation": ["Dubbel glas"], "heating": ["dummy"], "ownership": ["dummy"], "exteriors": ["dummy"], "parking": ["dummy"], "neighborhood_name": ["dummy"], "date_list": ["30 juni 2023"], "date_sold": ["13 juli 2023"], "term": ["13 dagen"], "price_sold": ["€ 500.000 k.k."], "last_ask_price": ["€ 500.000 kosten koper"], "last_ask_price_m2": ["dummy"], "city": ["utrecht"], "log_id": ["dummy"], "photo": ['dummy'] } return pd.DataFrame(data) class TestPreprocessData: def test_is_past_true(self, input_data): df = preprocess_data(df=input_data, is_past=True) assert df.shape == (1, 17) assert df["house_type"].item() == "appartement" assert df["price"].item() == 500000 assert df["room"].item() == 4 assert df["bedroom"].item() == 3 # assert df["term_days"].item() == 13 assert df["energy_label"].item() == ">A+"