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)