23 lines
684 B
Python
23 lines
684 B
Python
from finn_eiendom.models import EiendomUnit, FinnAd
|
|
from finn_eiendom.scoring import classify_ad, score_ad
|
|
|
|
|
|
def test_score_ad_and_classify():
|
|
ad = FinnAd(
|
|
finnkode="462400360",
|
|
url="https://www.finn.no/realestate/homes/ad.html?finnkode=462400360",
|
|
title="Flott 3-roms i Ferner",
|
|
)
|
|
unit = EiendomUnit(
|
|
unit_code="c-gxw-xmyum-s2a",
|
|
estimated_selling_price=7650000,
|
|
listing_price=7200000,
|
|
property_type="APARTMENT",
|
|
usable_area=77,
|
|
rooms=4,
|
|
)
|
|
scores = score_ad(ad, unit, [])
|
|
assert scores["market_position"] >= 0
|
|
categories = classify_ad(scores)
|
|
assert isinstance(categories, list)
|