def test_last_or_else(self):
# given
sut_1 = stream(range(5))
sut_2 = stream([])
# when
result_1 = sut_1.last_or_else()
result_2 = sut_2.last_or_else()
result_3 = sut_2.last_or_else("nothing")
# then
assert result_1 == 4
assert result_2 is None
assert result_3 == "nothing"
def test_for_each(self):
# given
class Person:
def __init__(self, name):
self.name = name
def upper_name(self):
self.name = self.name.upper()
sut = stream([
Person("John"),
Person("Paul"),
Person("George"),
Person("rinGo")
])
# when
sut.for_each(lambda x: x.upper_name())
result = sut.to_list()
# then
assert result[0].name == "JOHN"
assert result[1].name == "PAUL"
assert result[2].name == "GEORGE"
assert result[3].name == "RINGO"
def test_filter_to_list(self):
# given
sut = stream(range(10))
# when
result = sut.filter(lambda x: (x + 1) % 2 == 0).to_list()
# then
assert result == [1, 3, 5, 7, 9]
def test_take(self):
# given
sut = stream(range(5))
# when
result = sut.take(3).to_list()
# then
assert result == [0, 1, 2]
def test_filter_take(self):
# given
sut = stream(range(10))
# when
result = sut.filter(lambda x: (x + 1) % 2 == 0).take(20).to_list()
# then
self.assertEqual(result, [1, 3, 5, 7, 9])
def test_size(self):
# given
sut = stream(range(5000))
# when
result = sut.size()
# then
assert result == 5000
def test_map_to_list(self):
# given
sut = stream(range(5)).map(lambda x: x + 1)
# when
result = sut.to_list()
# then
self.assertEqual(result, [1, 2, 3, 4, 5])
def test_filter_chain_to_list(self):
# given
sut = stream(range(10))
# when
result = sut.filter(lambda x: x % 2 == 0).reverse().to_list()
# then
self.assertEqual(result, [8, 6, 4, 2, 0])
def test_size(self):
# given
sut = stream(range(50)).map(lambda x: x * 0.5)
# when
result = sut.size()
# then
self.assertEqual(result, 50)
def test_size(self):
# given
sut = stream(range(5000))
# when
result = sut.size()
# then
self.assertEqual(result, 5000)
def test_to_list(self):
# given
sut = stream(range(5))
# when
result = sut.reverse().to_list()
# then
self.assertEqual(result, [4, 3, 2, 1, 0])
def test_reduce(self):
# given
sut = stream(range(5))
# when
result = sut.reduce(lambda x, y: "%s%s" % (x, y))
# then
assert result == "01234"
def test_max(self):
# given
sut = stream(range(5))
# when
result = sut.max()
# then
assert result == 4
def test_map_filter(self):
# given
sut = stream(range(10))
# when
result = sut.map(lambda x: x // 2).filter(
lambda x: x % 2 == 0).to_string()
# then
assert result == "0, 0, 2, 2, 4, 4"
def test_size(self):
# given
sut = stream(range(50)).filter(lambda x: x % 2 == 0)
# when
result_1 = sut.size()
result_2 = sut.size() # test cached result
# then
self.assertEqual(result_1, 25)
self.assertEqual(result_2, 25)
def test_size(self):
# given
sut = stream(range(50)).filter(lambda x: x % 2 == 0)
# when
result_1 = sut.size()
result_2 = sut.size() # test cached result
# then
assert result_1 == 25
assert result_2 == 25
def test_to_string(self):
# given
sut = stream(range(5))
# when
result_1 = sut.to_string()
result_2 = sut.to_string("-")
# then
assert result_1 == "0, 1, 2, 3, 4"
assert result_2 == "0-1-2-3-4"
