def test_simple_one_dim(self): # Testing t-value and meandiff # First with paired setup paired_study = True stats = ptests.mean_difference_permtest( self.simple_data, self.simple_data_n1, self.simple_data_n2, paired_study, "all", seed=1, n_cpus=1 ) self.assertEqual(stats[0], self.simple_data_p_paired) self.assertAlmostEqual(stats[1], self.simple_data_meandiff) stats = ptests.t_value_permtest( self.simple_data, self.simple_data_n1, self.simple_data_n2, paired_study, "all", seed=1, n_cpus=1 ) self.assertEqual(stats[0], self.simple_data_p_paired) self.assertAlmostEqual(stats[1], self.simple_data_t_value_paired) # Same stuff with a unpaired setup paired_study = False stats = ptests.mean_difference_permtest( self.simple_data, self.simple_data_n1, self.simple_data_n2, paired_study, 10 ** 4.0, n_cpus=1, seed=10 ) self.assertTrue(np.abs(stats[0] - self.simple_data_p_unpaired) < 0.01) self.assertAlmostEqual(stats[1], self.simple_data_meandiff) stats = ptests.t_value_permtest( self.simple_data, self.simple_data_n1, self.simple_data_n2, paired_study, 10 ** 4.0, seed=1, n_cpus=1 ) self.assertTrue(np.abs(stats[0] - self.simple_data_p_unpaired) < 0.01) self.assertAlmostEqual(stats[1], self.simple_data_t_value_unpaired)
def test_two_dim_and_parallel(self): # test two dim data with just the basic mean difference sdn = 10 two_dim_data = np.array([self.simple_data for i in range(sdn)]).T # note the transpose!!! two_dim_pvals = np.ones((sdn,)) * self.simple_data_p_paired two_dim_meandiffs = np.ones((sdn,)) * self.simple_data_meandiff paired_study = True stats = ptests.mean_difference_permtest( two_dim_data, self.simple_data_n1, self.simple_data_n2, paired_study, "all", seed=1, n_cpus=1 ) self.as_arr_alm_eq(stats[0], two_dim_pvals) self.as_arr_alm_eq(stats[1], two_dim_meandiffs) stats = ptests.mean_difference_permtest( two_dim_data, self.simple_data_n1, self.simple_data_n2, paired_study, "all", seed=1, n_cpus=5 ) self.as_arr_alm_eq(stats[0], two_dim_pvals) self.as_arr_alm_eq(stats[1], two_dim_meandiffs)