コード例 #1
0
ファイル: test_topo.py プロジェクト: dynaryu/Wind_multipliers
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)
コード例 #2
0
ファイル: test_topo.py プロジェクト: dynaryu/Wind_multipliers
    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)