def test2(self):
"""test 02: order_by test"""
pq = PyLINQ(data)
self.assertEqual(
pq.where(lambda it: it["size"] < 10)
.order_by(lambda it: it["size"])
.select(lambda it: (it["name"], it["size"])).items(),
[("item3", 6), ("item1", 8)])
def test3(self):
"""test 03: count test with generator"""
pq = PyLINQ(iter(data))
self.assertEqual(
pq.where(lambda it: it["size"] < 10)
.count(), 2)
self.assertEqual(
pq.where(lambda it: it["class"] == "classB")
.count(), 1)
def test1(self):
"""test 01: where/select test"""
pq = PyLINQ(data)
self.assertEqual(
pq.where(lambda it: it["class"] == "classA")
.select(lambda it: {"name": it["name"], "size": it["size"]}).items(),
[{"name": "item1", "size": 8},
{"name": "item2", "size": 10}])
