mirror of
https://github.com/Hestia-Homes/survey-extraction.git
synced 2026-06-08 11:17:29 +00:00
32 lines
No EOL
1.4 KiB
Python
32 lines
No EOL
1.4 KiB
Python
# A very simple script to read a presite note and load to a postgres database
|
|
import os
|
|
from etl.surveyedData.surveryedData import surveyedDataProcessor
|
|
from etl.db.db import get_db_session, init_db
|
|
from etl.transform.preSiteNoteTypes import AssessorInfo
|
|
import pytest
|
|
from etl.jjc_old_lewis_manual_way_ import work_out_total_floor_area
|
|
|
|
@pytest.fixture(scope="module")
|
|
def pre_site_note_path():
|
|
"""Fixture for the path to the example pre-site note."""
|
|
return os.path.join(os.getcwd(), "example_data", "pre_site_note.pdf")
|
|
|
|
@pytest.fixture(scope="module")
|
|
def local_survey(pre_site_note_path):
|
|
"""Fixture to create a surveyedDataProcessor instance."""
|
|
return surveyedDataProcessor("122 Fake Street", [pre_site_note_path])
|
|
|
|
def test_assessor_accreditation(local_survey):
|
|
"""Test that the assessor accreditation number is correct."""
|
|
assert local_survey.pre_site_note.assessor_information.accreditation_number == "QUID211435"
|
|
|
|
|
|
def test_floor_area_calculator(local_survey):
|
|
# Floor area is important to work out invoice, make a test to work out invoice correctly
|
|
area = work_out_total_floor_area(local_survey.pre_site_note)
|
|
assert area == ('73-97m', 91)
|
|
|
|
|
|
def test_address_and_post_code(local_survey):
|
|
assert local_survey.pre_site_note.survey_information.address == "10 Turnberry Close, ST. LEONARDS-ON-SEA"
|
|
assert local_survey.pre_site_note.survey_information.postcode == "TN38 0WL" |