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)
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)