Coverage for src / tests / test_take.py: 100%

1import remedapy as R 

2 

3 

4class TestTake: 

5 def test_data_first(self): 

6 # R.take(array, n); 

7 assert list(R.take([1, 2, 3, 4, 3, 2, 1], 3)) == [1, 2, 3] 

8 

9 def test_data_last(self): 

10 # R.take(n)(array); 

11 assert R.pipe([1, 2, 3, 4, 3, 2, 1], R.take(3), list) == [1, 2, 3]