def test_r0p5_pipe(self): subsample.subsample(self.pipefile, self.outfile, subsample_rate=0.5, seed=self.seed) result = self.outfile.getvalue() benchmark = 'name|age|weight\r\nian|1|11\r\nchang|3|33\r\n' self.assertEqual(result, benchmark)
def test_r0p5_comma(self): subsample.subsample(self.commafile, self.outfile, subsample_rate=0.5, seed=self.seed) result = self.outfile.getvalue() benchmark = 'name,age,weight\r\nian,1,11\r\nchang,3,33\r\n' self.assertEqual(result, benchmark)
def test_r0p5_keyname_comma(self): subsample.subsample(self.longfile, self.outfile, subsample_rate=0.5, key_column='name', seed=self.seed) result = self.outfile.getvalue() benchmark = 'name,age,weight\r\nian,1,11\r\nian,1b,11b\r\nchang,3,33'\ '\r\nchang,3b,33b\r\n' self.assertEqual(result, benchmark)