class TestDummyOutput(ParametrizedTestCase):
"""
**This is a placeholder class**, currently used to simulate comparison of NETCDF output.
This test should be updated once NETCDF output is established for topomult.
"""
def test_TopoOutputNC(self):
"""
**This is a placeholder function**, currently used to simulate comparison of NETCDF output.
This test should be updated once NETCDF output is established for topomult.
Should also add loop for 8 directions.
"""
self.test_paths = TestPaths()
self.test_paths.initTopoTestPaths(self.param)
expectedArray = read_file.readNetCDF(os.path.join(self.test_paths.TOPO_TEST_EXPECTED_OUTPUT_DIR,
'netcdf_test_file.nc'),"e150s35demsCl")
actualArray = read_file.readNetCDF(os.path.join(self.test_paths.TOPO_TEST_ACTUAL_OUTPUT_DIR,
'netcdf_test_file.nc'), "e150s35demsCl")
# Note: allClose() will return False if has NaN
result = np.allclose(expectedArray, actualArray)
self.assertTrue(result)
class TestOutput(ParametrizedTestCase):
"""
Test Class used to run a unit test comparison of expected against actual results outputted by topomult.py
:param ParametrizedTestCase: Parametrized subclass of unittest.TestCase
"""
def test_TopoOutput(self):
"""
Runs current version of topomult.py, compares DEM output with expected for each direction using np.allClose()
"""
self.test_paths = TestPaths()
self.test_paths.initTopoTestPaths(self.param)
args = ['mpirun',
'-mca', 'btl', '^openib', # don't check for inifiniband
'-np', '8',
'python', os.path.join(self.test_paths.TOPO_MULTI_DIR, 'topomult.py'),
'-i', os.path.join(self.test_paths.TOPO_TEST_INPUT_DIR, 'dem.asc'),
'-o', self.test_paths.TOPO_TEST_ACTUAL_OUTPUT_DIR]
subprocess.call(args)
# need to loop through all directions..
for aDirection in DIRECTIONS:
print 'Dir:', aDirection
expectedArray = read_file.readASC(os.path.join(self.test_paths.TOPO_TEST_EXPECTED_OUTPUT_DIR,
'mh_'+ aDirection + '_smooth.asc'))
actualArray = read_file.readASC(os.path.join(self.test_paths.TOPO_TEST_ACTUAL_OUTPUT_DIR,
'mh_'+ aDirection + '_smooth.asc'))
# ! allClose() will return False if NaN is detected
result = np.allclose(expectedArray, actualArray)
print "\n",aDirection, "test =", result
self.assertTrue(result)