Fix failing tests again

This commit is contained in:
Nitzan Pomerantz
2025-10-30 20:54:18 +02:00
parent 9dbfca459b
commit 6632ac1669
2 changed files with 20 additions and 18 deletions
+5 -7
View File
@@ -96,15 +96,13 @@ def filter_deals_by_criteria(
# This allows "דירה" to match "דירת גג", "דירה בבניין", etc.
if property_type_normalized.endswith('ה'):
property_type_variant = property_type_normalized[:-1] + 'ת'
if property_type_variant in deal_type_normalized:
# Found variant match, continue to next filter
pass
elif property_type_normalized in deal_type_normalized:
# Found original match, continue to next filter
pass
else:
if property_type_variant not in deal_type_normalized and property_type_normalized not in deal_type_normalized:
# No match found for either variant
continue
else:
# For non-ה endings, do substring match
if property_type_normalized not in deal_type_normalized:
continue
# Room count filter
if min_rooms is not None or max_rooms is not None:
+15 -11
View File
@@ -167,10 +167,11 @@ class TestFilterDealsByCriteria:
min_price=1000000.0,
max_price=1500000.0,
)
# Should match only objectid=1 (דירה with 3 rooms, price 1M)
# objectid=2 is "דירת גג" not "דירה", so filtered out
assert len(result) == 1
assert set(d.objectid for d in result) == {1}
# Should match objectid=1 and objectid=2
# objectid=1: "דירה", 3 rooms, 1M price ✓
# objectid=2: "דירת גג" (matches "דירה" variant), 4 rooms, 1.5M price ✓
assert len(result) == 2
assert set(d.objectid for d in result) == {1, 2}
def test_filter_using_dealfilters_model(self, sample_deals):
"""Test filtering using DealFilters model."""
@@ -178,10 +179,11 @@ class TestFilterDealsByCriteria:
property_type="דירה", min_rooms=3.0, max_rooms=4.0
)
result = filter_deals_by_criteria(sample_deals, filters=filters)
# Should match only objectid=1 (דירה with 3 rooms)
# objectid=2 is "דירת גג" not exact match
assert len(result) == 1
assert set(d.objectid for d in result) == {1}
# Should match objectid=1 and objectid=2
# objectid=1: "דירה", 3 rooms ✓
# objectid=2: "דירת גג" (matches "דירה" variant), 4 rooms ✓
assert len(result) == 2
assert set(d.objectid for d in result) == {1, 2}
def test_filter_using_dict(self, sample_deals):
"""Test filtering using dict (converted to DealFilters)."""
@@ -191,9 +193,11 @@ class TestFilterDealsByCriteria:
"max_rooms": 4.0,
}
result = filter_deals_by_criteria(sample_deals, filters=filters_dict)
# Should match only objectid=1
assert len(result) == 1
assert set(d.objectid for d in result) == {1}
# Should match objectid=1 and objectid=2
# objectid=1: "דירה", 3 rooms ✓
# objectid=2: "דירת גג" (matches "דירה" variant), 4 rooms ✓
assert len(result) == 2
assert set(d.objectid for d in result) == {1, 2}
def test_individual_params_override_filters_model(self, sample_deals):
"""Test that individual parameters override filters model."""