HomeHarvest/tests/test_realtor.py

122 lines
3.1 KiB
Python
Raw Normal View History

2023-09-15 20:58:54 -07:00
from homeharvest import scrape_property
2023-09-18 20:59:49 -07:00
from homeharvest.exceptions import (
InvalidListingType,
NoResultsFound,
)
2023-09-15 20:58:54 -07:00
2023-10-04 18:25:01 -07:00
def test_realtor_pending_or_contingent():
pending_or_contingent_result = scrape_property(
2023-10-09 09:00:36 -07:00
location="Surprise, AZ", listing_type="pending"
2023-10-04 18:25:01 -07:00
)
2023-10-09 09:00:36 -07:00
regular_result = scrape_property(location="Surprise, AZ", listing_type="for_sale")
2023-10-04 18:25:01 -07:00
2023-10-04 20:33:21 -07:00
assert all(
[
result is not None
for result in [pending_or_contingent_result, regular_result]
]
)
2023-10-04 18:25:01 -07:00
assert len(pending_or_contingent_result) != len(regular_result)
def test_realtor_pending_comps():
pending_comps = scrape_property(
location="2530 Al Lipscomb Way",
radius=5,
past_days=180,
listing_type="pending",
)
for_sale_comps = scrape_property(
location="2530 Al Lipscomb Way",
radius=5,
past_days=180,
listing_type="for_sale",
)
sold_comps = scrape_property(
location="2530 Al Lipscomb Way",
radius=5,
past_days=180,
listing_type="sold",
)
results = [pending_comps, for_sale_comps, sold_comps]
assert all([result is not None for result in results])
#: assert all lengths are different
assert len(set([len(result) for result in results])) == len(results)
2023-10-02 13:58:47 -07:00
def test_realtor_comps():
result = scrape_property(
location="2530 Al Lipscomb Way",
radius=0.5,
2023-10-04 21:35:14 -07:00
past_days=180,
listing_type="sold",
2023-10-02 13:58:47 -07:00
)
2023-10-03 15:05:17 -07:00
assert result is not None and len(result) > 0
def test_realtor_last_x_days_sold():
days_result_30 = scrape_property(
2023-10-04 21:35:14 -07:00
location="Dallas, TX", listing_type="sold", past_days=30
2023-10-03 15:05:17 -07:00
)
days_result_10 = scrape_property(
2023-10-04 21:35:14 -07:00
location="Dallas, TX", listing_type="sold", past_days=10
2023-10-03 15:05:17 -07:00
)
assert all(
[result is not None for result in [days_result_30, days_result_10]]
) and len(days_result_30) != len(days_result_10)
2023-10-02 13:58:47 -07:00
2023-10-03 23:33:53 -07:00
def test_realtor_single_property():
results = [
scrape_property(
location="15509 N 172nd Dr, Surprise, AZ 85388",
listing_type="for_sale",
),
scrape_property(
location="2530 Al Lipscomb Way",
listing_type="for_sale",
),
2023-10-03 23:33:53 -07:00
]
assert all([result is not None for result in results])
2023-09-15 20:58:54 -07:00
def test_realtor():
results = [
scrape_property(
location="2530 Al Lipscomb Way",
listing_type="for_sale",
),
scrape_property(
2023-10-03 22:21:16 -07:00
location="Phoenix, AZ", listing_type="for_rent"
), #: does not support "city, state, USA" format
scrape_property(
2023-10-03 22:21:16 -07:00
location="Dallas, TX", listing_type="sold"
), #: does not support "city, state, USA" format
2023-10-03 22:21:16 -07:00
scrape_property(location="85281"),
]
2023-09-15 20:58:54 -07:00
assert all([result is not None for result in results])
bad_results = []
try:
bad_results += [
scrape_property(
location="abceefg ju098ot498hh9",
listing_type="for_sale",
)
]
2023-10-04 12:04:05 -07:00
except (InvalidListingType, NoResultsFound):
assert True
assert all([result is None for result in bad_results])