Coverage for tests/test_pipable.py: 100%

1from pipable import Pipe 

2import pytest 

3 

4 

5# == fixture == 

6def kebab(*args): 

7 return "-".join(args) 

8 

9 

10# == test == 

11def test_builtn_with_single_arg(): 

12 list_ = Pipe(list) 

13 assert "ab" | list_ == ["a", "b"] 

14 

15 

16def test_precedent_assign_to_last_pos_args(): 

17 assert "c" | Pipe(kebab) == "c" 

18 assert "c" | Pipe(kebab, "a") == "a-c" 

19 assert "c" | Pipe(kebab, "a", "b") == "a-b-c" 

20 

21 

22def test_assign_1st_arg(): 

23 base2 = Pipe(pow, 2) 

24 assert 3 | base2 == 8 

25 

26 

27def test_raise_pos_and_keyword_arg_conflict(): 

28 """ 

29 precedent will append to the positional argument list 

30 ∴ partial can't assign 1st arg as keyword 

31 """ 

32 base2 = Pipe(pow, base=2) 

33 with pytest.raises(Exception): 

34 3 | base2 

35 

36 

37def test_decorator_with_single_arg(): 

38 @Pipe 

39 def hi(name): 

40 return f"hi {name}" 

41 

42 assert "May" | hi == "hi May" 

43 

44 

45def test_decorator_with_multiple_arg(): 

46 @Pipe 

47 def power(base, exp): 

48 return base**exp 

49 

50 assert 3 | power(2) == 8 

51 assert 3 | power(exp=2) == 9 

52 # same reason with test_raise_pos_and_keyword_arg_conflict 

53 with pytest.raises(Exception): 

54 3 | power(base=2) 

55 

56 

57def test_wrap_with_partial(): 

58 base2 = Pipe(lambda x: pow(2, x)) 

59 assert 3 | base2 == 8 

60 

61 

62def test_variable_positional_args(capsys): 

63 print_ = Pipe(print, "a", "b") 

64 "c" | print_ 

65 stdout: str = capsys.readouterr().out 

66 assert stdout.splitlines()[-1] == "a b c" 

67 

68 

69def test_prepend_args_with_wrapper(): 

70 def wrapper(first, others): 

71 return kebab(first, *others) 

72 

73 assert "c" | Pipe(wrapper, others=["a", "b"]) == "c-a-b" 

74 

75 

76def test_decorator_prepend_args(): 

77 @Pipe 

78 def wrapper(first, args): 

79 return kebab(first, *args) 

80 

81 assert "c" | wrapper(args=["a", "b"]) == "c-a-b" 

82 

83 

84def test_reassign_with_wrapper(): 

85 def kebab(*args): 

86 return "-".join(args) 

87 

88 def wrapper(first, others): 

89 return kebab(first, *others) 

90 

91 # works 

92 assert "a" | Pipe(wrapper, others=["b", "c"]) == "a-b-c" 

93 

94 # works with other var name 

95 kebab_ = Pipe(wrapper, others=["b", "c"]) 

96 assert "a" | kebab_ == "a-b-c" 

97 

98 # not work, ∵ original name already re-assigned when wrapper is being invoked 

99 # it become a Pipe object instead 

100 kebab = Pipe(wrapper, others=["b", "c"]) 

101 result = "a" | kebab 

102 assert result != "a-b-c" 

103 assert isinstance(result, Pipe)