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)
