from linq import linq
print(
linq(open(
"words",
"r").readlines()).Select(lambda line: line.split()).Flatten().GroupBy(
lambda word: word).Select(lambda item: (item[0], len(item[1]))).
OrderBy(lambda item: item[1]).Select(
lambda item: item[0] + ": " + str(item[1])).ToList())
from linq import linq
def fib_gen():
a, b = 1, 1
while True:
yield a
a, b = b, a + b
print (linq(fib_gen()) \
.Where(lambda x: x % 3 == 0) \
.Select(lambda x: x ** 2 if x % 2 else x) \
.Take(5) \
.ToList())
def test_linq_to_dict():
data = [(1, "a"), (2, "b")]
expected = {1: "a", 2: "b"}
actual = linq(data).to_dict(key=lambda x: x[0], value=lambda x: x[1])
assert actual == expected
def test_linq_single_raises_value_error():
data = [1, 2, 3, 4]
with pytest.raises(ValueError):
linq(data).single()
def test_linq_contains():
data = [1, 3, 4, 5]
l = linq(data)
assert all(l.contains(x) for x in data)
def test_linq_count():
data = [1, 2, 3, 4]
expected = 4
actual = linq(data).count()
assert actual == expected
def test_linq_iter():
data = [1, 2, 3, 4]
actual = list(x for x in linq(data))
expected = data
assert actual == expected
def test_linq_skip_while():
data = [1, 2, 3, 4]
actual = linq(data).skip_while(lambda x: x < 3).to_list()
expected = [3, 4]
assert actual == expected
def test_select_many():
data = [1, 2, 3]
expected = [1, 1, 2, 2, 3, 3]
actual = linq(data).select_many(lambda x: [x] * 2).to_list()
assert actual == expected
def test_linq_skip():
data = [1, 2, 3, 4]
expected = [3, 4]
actual = linq(data).skip(2).to_list()
assert actual == expected
def test_linq_take():
data = [1, 2, 3, 4]
expected = [1, 2]
actual = linq(data).take(2).to_list()
assert actual == expected
def test_linq_where():
data = [1, 2, 3, 4]
expected = [2, 4]
actual = linq(data).where(lambda x: x % 2 == 0).to_list()
assert actual == expected
def test_linq_select():
data = [1, 2]
expected = [2, 4]
actual = linq(data).select(lambda x: x * 2).to_list()
assert actual == expected
