def test_mst_equal(self): ifgs = common.small_data_setup(datafiles=self.dest_paths) mst_original_p = mst.mst_boolean_array(ifgs) ifgs_s = common.small_data_setup(datafiles=self.dest_paths_s) mst_original_s = mst.mst_boolean_array(ifgs_s) np.testing.assert_array_equal(self.mst, mst_original_p) np.testing.assert_array_equal(self.mst, mst_original_s) np.testing.assert_array_equal(self.mst, self.mst_p)
def setUpClass(cls): cls.ifgs = common.small_data_setup() cls.params = default_params() cls.mstmat = mst.mst_boolean_array(cls.ifgs) r_dist = covariance.RDist(cls.ifgs[0])() cls.maxvar = [covariance.cvd(i, cls.params, r_dist)[0] for i in cls.ifgs] cls.vcmt = covariance.get_vcmt(cls.ifgs, cls.maxvar)
def test_mst_multiprocessing(self): self.params[cf.PARALLEL] = True original_mst = mst.mst_boolean_array(self.ifgs) parallel_mst = mst.mst_parallel(self.ifgs, params=self.params) np.testing.assert_array_equal(original_mst, parallel_mst)