diff --git a/model_data/tests/test_property.py b/model_data/tests/test_property.py index 6ae90316..964bda00 100644 --- a/model_data/tests/test_property.py +++ b/model_data/tests/test_property.py @@ -16,7 +16,8 @@ mock_epc_response = { "walls-description": "Walls Description", "windows-description": "Windows Description", "mainheat-description": "Main Heating Description", - "hotwater-description": "Hot Water Description" + "hotwater-description": "Hot Water Description", + "transaction-type": "rental" }, { "inspection-date": "2023-05-01", @@ -25,7 +26,8 @@ mock_epc_response = { "walls-description": "Walls Description", "windows-description": "Windows Description", "mainheat-description": "Main Heating Description", - "hotwater-description": "Hot Water Description" + "hotwater-description": "Hot Water Description", + "transaction-type": "rental" } ] } @@ -35,19 +37,22 @@ mock_epc_response_dupe = { { 'inspection-date': '2023-06-01', 'some-other-key': 'some-value', 'roof-description': 'Roof Description', 'walls-description': 'Walls Description', 'windows-description': 'Windows Description', - 'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description' + 'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description', + "transaction-type": "rental" }, { 'inspection-date': '2023-05-01', 'some-other-key': 'some-other-value', 'roof-description': 'Roof Description', 'walls-description': 'Walls Description', 'windows-description': 'Windows Description', 'mainheat-description': 'Main Heating Description', - 'hotwater-description': 'Hot Water Description' + 'hotwater-description': 'Hot Water Description', + "transaction-type": "rental" }, { 'inspection-date': '2023-06-01', 'some-other-key': 'duplicate-date', 'roof-description': 'Roof Description', 'walls-description': 'Walls Description', 'windows-description': 'Windows Description', - 'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description' + 'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description', + "transaction-type": "rental" } ] }