def test_get_distance_boxplot_data_two_groups(self): dm = skbio.DistanceMatrix( [[0.00, 0.12, 0.13, 0.14, 0.15], [0.12, 0.00, 0.22, 0.23, 0.24], [0.13, 0.22, 0.00, 0.31, 0.32], [0.14, 0.23, 0.31, 0.00, 0.44], [0.15, 0.24, 0.32, 0.44, 0.00]], ids=['s1', 's2', 's3', 's4', 's5']) groupings = collections.OrderedDict([('g1', ['s1', 's2']), ('g2', ['s3', 's4', 's5'])]) obs = _get_distance_boxplot_data(dm, 'g1', groupings) exp_data = [[0.12], [0.13, 0.14, 0.15, 0.22, 0.23, 0.24]] exp_labels = ['g1 (n=1)', 'g2 (n=6)'] self.assertEqual(obs[0], exp_data) self.assertEqual(obs[1], exp_labels)
def test_get_distance_boxplot_data_between_order_retained(self): dm = skbio.DistanceMatrix([[0.00, 0.12, 0.13, 0.14, 0.15], [0.12, 0.00, 0.22, 0.23, 0.24], [0.13, 0.22, 0.00, 0.31, 0.32], [0.14, 0.23, 0.31, 0.00, 0.44], [0.15, 0.24, 0.32, 0.44, 0.00]], ids=['s1', 's2', 's3', 's4', 's5']) groupings = collections.OrderedDict( [('g1', ['s1', 's2']), ('g3', ['s4']), ('g2', ['s3', 's5'])]) obs = _get_distance_boxplot_data(dm, 'g1', groupings) exp_data = [[0.12], [0.14, 0.23], [0.13, 0.15, 0.22, 0.24]] exp_labels = ['g1 (n=1)', 'g3 (n=2)', 'g2 (n=4)'] self.assertEqual(obs[0], exp_data) self.assertEqual(obs[1], exp_labels)
def test_get_distance_boxplot_data_within_always_first(self): dm = skbio.DistanceMatrix( [[0.00, 0.12, 0.13, 0.14, 0.15], [0.12, 0.00, 0.22, 0.23, 0.24], [0.13, 0.22, 0.00, 0.31, 0.32], [0.14, 0.23, 0.31, 0.00, 0.44], [0.15, 0.24, 0.32, 0.44, 0.00]], ids=['s1', 's2', 's3', 's4', 's5']) groupings = collections.OrderedDict([('g2', ['s3', 's4', 's5']), ('g1', ['s1', 's2'])]) obs = _get_distance_boxplot_data(dm, 'g1', groupings) exp_data = [[0.12], [0.13, 0.14, 0.15, 0.22, 0.23, 0.24]] exp_labels = ['g1 (n=1)', 'g2 (n=6)'] exp_summary = [('s2', 's1', 'g1', 'g1', 0.12), ('s1', 's3', 'g1', 'g2', 0.13), ('s1', 's4', 'g1', 'g2', 0.14000000000000001), ('s1', 's5', 'g1', 'g2', 0.14999999999999999), ('s2', 's3', 'g1', 'g2', 0.22), ('s2', 's4', 'g1', 'g2', 0.23000000000000001), ('s2', 's5', 'g1', 'g2', 0.23999999999999999)] self.assertEqual(obs[0], exp_data) self.assertEqual(obs[1], exp_labels) self.assertEqual(obs[2], exp_summary)
def test_get_distance_boxplot_data_within_always_first(self): dm = skbio.DistanceMatrix([[0.00, 0.12, 0.13, 0.14, 0.15], [0.12, 0.00, 0.22, 0.23, 0.24], [0.13, 0.22, 0.00, 0.31, 0.32], [0.14, 0.23, 0.31, 0.00, 0.44], [0.15, 0.24, 0.32, 0.44, 0.00]], ids=['s1', 's2', 's3', 's4', 's5']) groupings = collections.OrderedDict( [('g2', ['s3', 's4', 's5']), ('g1', ['s1', 's2'])]) obs = _get_distance_boxplot_data(dm, 'g1', groupings) exp_data = [[0.12], [0.13, 0.14, 0.15, 0.22, 0.23, 0.24]] exp_labels = ['g1 (n=1)', 'g2 (n=6)'] exp_summary = [('s2', 's1', 'g1', 'g1', 0.12), ('s1', 's3', 'g1', 'g2', 0.13), ('s1', 's4', 'g1', 'g2', 0.14000000000000001), ('s1', 's5', 'g1', 'g2', 0.14999999999999999), ('s2', 's3', 'g1', 'g2', 0.22), ('s2', 's4', 'g1', 'g2', 0.23000000000000001), ('s2', 's5', 'g1', 'g2', 0.23999999999999999)] self.assertEqual(obs[0], exp_data) self.assertEqual(obs[1], exp_labels) self.assertEqual(obs[2], exp_summary)