Skip to content

Commit

Permalink
Adding assertions about population group accessibility.
Browse files Browse the repository at this point in the history
  • Loading branch information
phargogh committed Jun 30, 2023
1 parent d49d797 commit a31b167
Showing 1 changed file with 22 additions and 1 deletion.
23 changes: 22 additions & 1 deletion tests/test_urban_nature_access.py
Original file line number Diff line number Diff line change
Expand Up @@ -569,7 +569,6 @@ def _assert_urban_nature(path, sum_value, min_value, max_value):
_assert_urban_nature(os.path.join(
output_dir, 'accessible_urban_nature_lucode_9_suffix.tif'),
7744116.974121094, 1567.57958984375, 12863.4619140625)
assert print(os.listdir(output_dir)) == 0

def test_split_population(self):
"""UNA: test split population optional module.
Expand Down Expand Up @@ -705,6 +704,28 @@ def test_radii_by_pop_group(self):
self.assertAlmostEqual(
expected_value, summary_feature.GetField(fieldname))

def _assert_urban_nature(path, sum_value, min_value, max_value):
accessible_urban_nature_array = (
pygeoprocessing.raster_to_numpy_array(path))
valid_mask = ~utils.array_equals_nodata(
accessible_urban_nature_array,
urban_nature_access.FLOAT32_NODATA)
valid_pixels = accessible_urban_nature_array[valid_mask]
print(f"{numpy.sum(valid_pixels)}, {numpy.min(valid_pixels)}, "
f"{numpy.max(valid_pixels)}")
self.assertAlmostEqual(numpy.sum(valid_pixels), sum_value)
self.assertAlmostEqual(numpy.min(valid_pixels), min_value)
self.assertAlmostEqual(numpy.max(valid_pixels), max_value)

output_dir = os.path.join(args['workspace_dir'], 'output')
_assert_urban_nature(os.path.join(
output_dir, 'accessible_urban_nature_to_pop_male.tif'),
6221004.412597656, 1171.7352294921875, 11898.0712890625)
_assert_urban_nature(os.path.join(
output_dir, 'accessible_urban_nature_to_pop_female.tif'),
6221004.412597656, 1171.7352294921875, 11898.0712890625)


def test_modes_same_radii_same_results(self):
"""UNA: all modes have same results when consistent radii.
Expand Down

0 comments on commit a31b167

Please sign in to comment.