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
)
