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