54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
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²"],
|
|
"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+"
|