def test_groupby(self): lst = _(self.seq) self.assertEqual( lst.groupby(self.evenf)._, [(0, [0, 2, 4, 6, 8]), (1, [1, 3, 5, 7, 9])] )
def test_distinct(self): lst = _(self.seq * 2) self.assertEqual(lst.distinct()._, list(set(self.seq)))
def test_over_top(self): lst = _(self.seq) self.assertEqual(lst.top(10000)._, sorted(self.seq, reverse=True))
def test_var(self): lst = _(self.seq) mean = sum(self.seq) / len(self.seq) self.assertAlmostEqual( lst.variance(), sum((x - mean)**2 for x in self.seq) / (len(self.seq) - 1))
def test_top(self): lst = _(self.seq) self.assertEqual(lst.top(2)._, sorted(self.seq, reverse=True)[:2])
def test_slice_start_end_skip(self): lst = _(self.seq) self.assertEqual(lst.slice(2, 8, 3)._, self.seq[2:8:3])
def test_flat_reduce(self): lst = _(self.seq) self.assertEqual(lst.reduce(self.sumf), sum(self.seq))
def test_fmap(self): lst = _(self.seq) self.assertEqual( lst.flatmap(self.flatmapf)._, reduce(self.sumf, map(self.flatmapf, self.seq)))
def test_over_top(self): lst = _(self.seq) self.assertEqual( lst.top(10000)._, sorted(self.seq, reverse=True) )
def test_one_top(self): lst = _(self.seq) self.assertEqual( lst.top()._[0], max(self.seq) )
def test_top(self): lst = _(self.seq) self.assertEqual( lst.top(2)._, sorted(self.seq, reverse=True)[:2] )
def test_sort(self): lst = _(self.seq) self.assertEqual( lst.sort(reverse=True)._, sorted(self.seq, reverse=True) )
def test_over_take(self): lst = _(self.seq) self.assertEqual( lst.take(3000)._, self.seq )
def test_take(self): lst = _(self.seq) self.assertEqual( lst.take(3)._, self.seq[:3] )
def test_slice_start_end_skip(self): lst = _(self.seq) self.assertEqual( lst.slice(2, 8, 3)._, self.seq[2:8:3] )
def test_size(self): lst = _(self.seq) self.assertEqual(lst.size(), len(self.seq))
def test_filter(self): lst = _(self.seq) self.assertEqual( lst.filter(self.filterf)._, filter(self.filterf, self.seq))
def test_sample(self): lst = _(self.seq) self.assertEqual( lst.sample(4).size(), 4 )
def test_dropwhile(self): lst = _(self.seq) self.assertEqual(lst.dropwhile(self.dropwhilef)._, self.seq[5:])
def test_distinct(self): lst = _(self.seq*2) self.assertEqual( lst.distinct()._, list(set(self.seq)) )
def test_slice_start(self): lst = _(self.seq) self.assertEqual(lst.slice(5)._, self.seq[:5])
def test_reduceby(self): lst = _([(1,1), (1,1), (1,1), (2,1)]) self.assertEqual( lst.reduceby(self.sumf)._, [(1, 3), (2,1)] )
def test_over_take(self): lst = _(self.seq) self.assertEqual(lst.take(3000)._, self.seq)
def test_size(self): lst = _(self.seq) self.assertEqual( lst.size(), len(self.seq) )
def test_slice_start_end(self): lst = _(self.seq) self.assertEqual( lst.slice(5, 8)._, self.seq[5:8] )
def test_mean(self): lst = _(self.seq) mean = sum(self.seq)/len(self.seq) self.assertAlmostEqual(lst.mean(), mean)
def test_groupby(self): lst = _(self.seq) self.assertEqual(lst.groupby(self.evenf), { 0: [0, 2, 4, 6, 8], 1: [1, 3, 5, 7, 9] })
def test_var(self): lst = _(self.seq) mean = sum(self.seq)/len(self.seq) self.assertAlmostEqual(lst.variance(), sum((x-mean)**2 for x in self.seq)/(len(self.seq)-1))
def test_sample_without_replacement(self): lst = _(self.seq) self.assertEqual(lst.sample_without_replacement(4).size(), 4)
def test_map(self): lst = _(self.seq) self.assertEqual( lst.map(self.mapf)._, map(self.mapf, self.seq) )
def test_reduceby(self): lst = _([(1, 1), (1, 1), (1, 1), (2, 1)]) self.assertEqual(lst.reduceby(self.sumf)._, [(1, 3), (2, 1)])
def test_filter(self): lst = _(self.seq) self.assertEqual( lst.filter(self.filterf)._, filter(self.filterf, self.seq) )
def test_map(self): lst = _(self.seq) self.assertEqual(lst.map(self.mapf)._, map(self.mapf, self.seq))
def test_filter_false(self): lst = _(self.seq) self.assertEqual( lst.filterfalse(self.filterf)._, filter(lambda x: not self.filterf(x), self.seq) )
def test_filter_false(self): lst = _(self.seq) self.assertEqual( lst.filterfalse(self.filterf)._, filter(lambda x: not self.filterf(x), self.seq))
def test_fmap(self): lst = _(self.seq) self.assertEqual( lst.flatmap(self.flatmapf)._, reduce(self.sumf, map(self.flatmapf, self.seq)) )
def test_takewhile(self): lst = _(self.seq) self.assertEqual(lst.takewhile(self.takewhilef)._, self.seq[:5])
def test_takewhile(self): lst = _(self.seq) self.assertEqual( lst.takewhile(self.takewhilef)._, self.seq[:5] )
def test_groupby(self): lst = _(self.seq) self.assertEqual( lst.groupby(self.evenf)._, [(0, [0, 2, 4, 6, 8]), (1, [1, 3, 5, 7, 9])])
def test_dropwhile(self): lst = _(self.seq) self.assertEqual( lst.dropwhile(self.dropwhilef)._, self.seq[5:] )
def test_union(self): lst = _(self.seq) self.assertEqual(lst.union(self.seq).union(self.seq)._, self.seq * 3)
def test_flat_reduce(self): lst = _(self.seq) self.assertEqual( lst.reduce(self.sumf), sum(self.seq) )
def test_slice_start_end(self): lst = _(self.seq) self.assertEqual(lst.slice(5, 8)._, self.seq[5:8])
def test_union(self): lst = _(self.seq) self.assertEqual( lst.union(self.seq).union(self.seq)._, self.seq * 3 )
def test_take(self): lst = _(self.seq) self.assertEqual(lst.take(3)._, self.seq[:3])
def test_slice_start(self): lst = _(self.seq) self.assertEqual( lst.slice(5)._, self.seq[:5] )
def test_sort(self): lst = _(self.seq) self.assertEqual( lst.sort(reverse=True)._, sorted(self.seq, reverse=True))
def test_mean(self): lst = _(self.seq) mean = sum(self.seq) / len(self.seq) self.assertAlmostEqual(lst.mean(), mean)
def test_one_top(self): lst = _(self.seq) self.assertEqual(lst.top()._[0], max(self.seq))
def test_sample_without_replacement(self): lst = _(self.seq) self.assertEqual( lst.sample_without_replacement(4).size(), 4 )