def test_proportion_plot_order_figure(self):
self.maxDiff = None
np.random.seed(0)
# tests for different ordering
fig, axes = plt.subplots(1, 2)
num_features = ['A', 'B']
denom_features = ['D', 'C']
ax1, ax2 = proportion_plot(self.table, self.metadata,
'groups', 'X', 'Y',
num_features, denom_features,
self.feature_metadata,
label_col='phylum', axes=axes)
res = np.vstack([l.get_xydata() for l in ax1.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, atol=1e-2, rtol=1e-2, verbose=True)
res = np.vstack([l.get_xydata() for l in ax2.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, atol=1e-2, rtol=1e-2, verbose=True)
res = [l._text for l in ax2.get_yticklabels()]
exp = ['p__bar', 'p__bar', 'p__far', 'p__tar']
self.assertListEqual(res, exp)
def test_proportion_plot(self):
np.random.seed(0)
num_features = ['A', 'B']
denom_features = ['C', 'D']
ax1, ax2 = proportion_plot(self.table,
self.metadata,
'groups',
'X',
'Y',
num_features,
denom_features,
self.feature_metadata,
label_col='phylum')
res = np.vstack([L.get_xydata() for L in ax1.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, verbose=True)
res = np.vstack([L.get_xydata() for L in ax2.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, verbose=True)
res = [L._text for L in ax2.get_yticklabels()]
exp = ['p__bar', 'p__bar', 'p__tar', 'p__far']
self.assertListEqual(res, exp)
def test_proportion_plot_order_figure(self):
self.maxDiff = None
np.random.seed(0)
# tests for different ordering
fig, axes = plt.subplots(1, 2)
num_features = ['A', 'B']
denom_features = ['D', 'C']
ax1, ax2 = proportion_plot(self.table,
self.metadata,
'groups',
'X',
'Y',
num_features,
denom_features,
self.feature_metadata,
label_col='phylum',
axes=axes)
res = np.vstack([L.get_xydata() for L in ax1.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, atol=1e-2, rtol=1e-2, verbose=True)
res = np.vstack([L.get_xydata() for L in ax2.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, atol=1e-2, rtol=1e-2, verbose=True)
res = [L._text for L in ax2.get_yticklabels()]
exp = ['p__bar', 'p__bar', 'p__far', 'p__tar']
self.assertListEqual(res, exp)
def test_proportion_plot_original_labels(self):
# tests for different ordering
fig, axes = plt.subplots(1, 2)
num_features = ['A', 'B']
denom_features = ['D', 'C']
ax1, ax2 = proportion_plot(self.table, self.metadata,
'groups', 'X', 'Y',
num_features, denom_features,
axes=axes)
res = [l._text for l in ax2.get_yticklabels()]
exp = ['A', 'B', 'D', 'C']
self.assertListEqual(res, exp)
def test_proportion_plot_original_labels(self):
# tests for different ordering
fig, axes = plt.subplots(1, 2)
num_features = ['A', 'B']
denom_features = ['D', 'C']
ax1, ax2 = proportion_plot(self.table,
self.metadata,
'groups',
'X',
'Y',
num_features,
denom_features,
axes=axes)
res = [L._text for L in ax2.get_yticklabels()]
exp = ['A', 'B', 'D', 'C']
self.assertListEqual(res, exp)
def test_proportion_plot(self):
np.random.seed(0)
num_features = ['A', 'B']
denom_features = ['C', 'D']
ax1, ax2 = proportion_plot(self.table, self.metadata,
'groups', 'X', 'Y',
num_features, denom_features,
self.feature_metadata,
label_col='phylum')
res = np.vstack([l.get_xydata() for l in ax1.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, verbose=True)
res = np.vstack([l.get_xydata() for l in ax2.get_lines()])
exp = np.array([0., 0., 1., 1., 2., 2., 3., 3.])
npt.assert_allclose(res[:, 1], exp, verbose=True)
res = [l._text for l in ax2.get_yticklabels()]
exp = ['p__bar', 'p__bar', 'p__tar', 'p__far']
self.assertListEqual(res, exp)