def test_cv_downsample(self):
npr1, ngs1 = ManagePriors.cross_validate_gold_standard(self.priors_data, self.gold_standard, None, 0.5, 42)
self.assertEqual(npr1.shape, ngs1.shape)
self.assertListEqual(npr1.index.tolist(), ngs1.index.tolist())
self.assertListEqual(npr1.columns.tolist(), ngs1.columns.tolist())
self.assertEqual((npr1 != 0).sum().sum(), (ngs1 != 0).sum().sum() * 0.5)
self.assertEqual((self.gold_standard != 0).sum().sum(), (ngs1 != 0).sum().sum())
pdt.assert_frame_equal(self.gold_standard, ngs1)
def test_cv_genes(self):
npr1, ngs1 = ManagePriors.cross_validate_gold_standard(
self.priors_data, self.gold_standard, 0, 0.5, 42)
self.assertEqual(npr1.shape, ngs1.shape)
self.assertEqual(len(npr1.index.intersection(ngs1.index)), 0)
pdt.assert_index_equal(npr1.columns, self.priors_data.columns)
pdt.assert_index_equal(ngs1.columns, self.gold_standard.columns)
npr2, ngs2 = ManagePriors.cross_validate_gold_standard(
self.priors_data, self.gold_standard, 0, 0.5, 43)
npr3, ngs3 = ManagePriors.cross_validate_gold_standard(
self.priors_data, self.gold_standard, 0, 0.5, 42)
pdt.assert_frame_equal(npr1, npr3)
pdt.assert_frame_equal(ngs1, ngs3)
with self.assertRaises(AssertionError):
pdt.assert_frame_equal(npr1, npr2)
with self.assertRaises(AssertionError):
pdt.assert_frame_equal(ngs1, ngs2)
def test_cv_tfs(self):
npr1, ngs1 = ManagePriors.cross_validate_gold_standard(
self.priors_data, self.gold_standard, 1, 0.5, 42)
self.assertEqual(npr1.shape, ngs1.shape)
self.assertEqual(len(npr1.columns.intersection(ngs1.columns)), 0)
def test_cv_genes(self):
npr1, ngs1 = ManagePriors.cross_validate_gold_standard(
self.priors_data, self.gold_standard, 0, 0.5, 42)
self.assertEqual(npr1.shape, ngs1.shape)
self.assertEqual(len(npr1.index.intersection(ngs1.index)), 0)
