def test_local_calc_existing_cached(self):
"""Recreation local PUD calculation on existing quadtree."""
from natcap.invest.recreation import recmodel_server
recreation_server = recmodel_server.RecModel(
self.resampled_data_path, 2005, 2014,
os.path.join(self.workspace_dir, 'server_cache'))
recreation_server = None
# This will not generate a new quadtree but instead load existing one
recreation_server = recmodel_server.RecModel(
self.resampled_data_path, 2005, 2014,
os.path.join(self.workspace_dir, 'server_cache'))
date_range = (numpy.datetime64('2005-01-01'),
numpy.datetime64('2014-12-31'))
poly_test_queue = Queue.Queue()
poly_test_queue.put(0)
poly_test_queue.put('STOP')
pud_poly_feature_queue = Queue.Queue()
recmodel_server._calc_poly_pud(
recreation_server.qt_pickle_filename,
os.path.join(SAMPLE_DATA, 'test_aoi_for_subset.shp'), date_range,
poly_test_queue, pud_poly_feature_queue)
# assert annual average PUD is the same as regression
self.assertEqual(83.2, pud_poly_feature_queue.get()[1][0])
def setUp(self):
"""Setup workspace and server."""
from natcap.invest.recreation import recmodel_server
self.workspace_dir = tempfile.mkdtemp()
self.recreation_server = recmodel_server.RecModel(
os.path.join(SAMPLE_DATA, 'sample_data.csv'), 2005, 2014,
os.path.join(self.workspace_dir, 'server_cache'))
def test_local_calc_poly_pud(self):
"""Recreation test single threaded local PUD calculation."""
from natcap.invest.recreation import recmodel_client
from natcap.invest.recreation import recmodel_server
recreation_server = recmodel_server.RecModel(
os.path.join(REGRESSION_DATA, 'sample_data.csv'),
2005, 2014, os.path.join(self.workspace_dir, 'server_cache'))
date_range = (
numpy.datetime64('2005-01-01'),
numpy.datetime64('2014-12-31'))
poly_test_queue = Queue.Queue()
poly_test_queue.put(0)
poly_test_queue.put('STOP')
pud_poly_feature_queue = Queue.Queue()
recmodel_server._calc_poly_pud(
recreation_server.qt_pickle_filename,
os.path.join(REGRESSION_DATA, 'test_aoi_for_subset.shp'),
date_range, poly_test_queue, pud_poly_feature_queue)
# assert annual average PUD is the same as regression
self.assertEqual(
53.3, pud_poly_feature_queue.get()[1][0])
def test_year_order(self):
"""Recreation ensure that end year < start year raise ValueError."""
from natcap.invest.recreation import recmodel_server
with self.assertRaises(ValueError):
# intentionally construct start year > end year
recmodel_server.RecModel(
self.resampled_data_path, 2014, 2005,
os.path.join(self.workspace_dir, 'server_cache'))
def test_local_aggregate_points(self):
"""Recreation test single threaded local AOI aggregate calculation."""
from natcap.invest.recreation import recmodel_client
from natcap.invest.recreation import recmodel_server
recreation_server = recmodel_server.RecModel(
os.path.join(REGRESSION_DATA, 'sample_data.csv'),
2005, 2014, os.path.join(self.workspace_dir, 'server_cache'))
if not os.path.exists(self.workspace_dir):
os.makedirs(self.workspace_dir)
aoi_path = os.path.join(REGRESSION_DATA, 'test_aoi_for_subset.shp')
basename = os.path.splitext(aoi_path)[0]
aoi_archive_path = os.path.join(
self.workspace_dir, 'aoi_zipped.zip')
with zipfile.ZipFile(aoi_archive_path, 'w') as myzip:
for filename in glob.glob(basename + '.*'):
myzip.write(filename, os.path.basename(filename))
# convert shapefile to binary string for serialization
zip_file_binary = open(aoi_archive_path, 'rb').read()
# transfer zipped file to server
date_range = (('2005-01-01'), ('2014-12-31'))
out_vector_filename = 'test_aoi_for_subset_pud.shp'
zip_result, workspace_id = (
recreation_server.calc_photo_user_days_in_aoi(
zip_file_binary, date_range, out_vector_filename))
# unpack result
result_zip_path = os.path.join(self.workspace_dir, 'pud_result.zip')
open(result_zip_path, 'wb').write(zip_result)
zipfile.ZipFile(result_zip_path, 'r').extractall(self.workspace_dir)
result_vector_path = os.path.join(
self.workspace_dir, out_vector_filename)
expected_vector_path = os.path.join(
REGRESSION_DATA, 'test_aoi_for_subset_pud.shp')
natcap.invest.pygeoprocessing_0_3_3.testing.assert_vectors_equal(
expected_vector_path, result_vector_path)
# ensure the remote workspace is as expected
workspace_zip_binary = recreation_server.fetch_workspace_aoi(
workspace_id)
out_workspace_dir = os.path.join(self.workspace_dir, 'workspace_zip')
os.makedirs(out_workspace_dir)
workspace_zip_path = os.path.join(out_workspace_dir, 'workspace.zip')
open(workspace_zip_path, 'wb').write(workspace_zip_binary)
zipfile.ZipFile(workspace_zip_path, 'r').extractall(out_workspace_dir)
natcap.invest.pygeoprocessing_0_3_3.testing.assert_vectors_equal(
aoi_path,
os.path.join(out_workspace_dir, 'test_aoi_for_subset.shp'))