a_adli/colouring-montreal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

name output_file with input_file

Browse Source
This commit is contained in:
Ed Chalstrey 2022-03-18 15:27:12 +00:00
parent 35f85ee2ed
commit a159c974dc
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_filter.py
View File

@ -6,7 +6,17 @@ from etl import filter_mastermap
def test_filter_mastermap(): def test_filter_mastermap():
"""Test that MasterMap CSV can be correctly filtered to include only buildings.""" """Test that MasterMap CSV can be correctly filtered to include only buildings."""
input_file = "tests/test_mastermap.gml.csv" # Test csv with one building and one non-building input_file = "tests/test_mastermap.gml.csv" # Test csv with one building and one non-building
output_file = input_file.replace('gml', 'filtered')
filter_mastermap(input_file) # creates test_mastermap.filtered.csv filter_mastermap(input_file) # creates test_mastermap.filtered.csv
with open('tests/test_mastermap.filtered.csv', newline='') as csvfile: with open(output_file, newline='') as csvfile:
csv_array = list(csv.reader(csvfile)) csv_array = list(csv.reader(csvfile))
assert len(csv_array) == 2 # assert that length is 2 because just one row after header assert len(csv_array) == 2 # assert that length is 2 because just one row after header
# def test_filter_mastermap_missing_type():
# """Test that MasterMap CSV can be correctly filtered when the polygon does not have a type specified."""
# input_file = "tests/test_mastermap_missing_type.gml.csv" # Test csv with one building and one non-building
# filter_mastermap(input_file) # creates test_mastermap.filtered.csv
# with open('tests/test_mastermap.filtered.csv', newline='') as csvfile:
# csv_array = list(csv.reader(csvfile))
# assert len(csv_array) == 2 # assert that length is 2 because just one row after header