Skip to content

Commit a0cbc20

Browse files
authored
refactor: Make code more simple in maclaurin_series (TheAlgorithms#7522)
1 parent f32f78a commit a0cbc20

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

maths/maclaurin_series.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,7 @@ def maclaurin_sin(theta: float, accuracy: int = 30) -> float:
5252
div = theta // (2 * pi)
5353
theta -= 2 * div * pi
5454
return sum(
55-
(((-1) ** r) * ((theta ** (2 * r + 1)) / factorial(2 * r + 1)))
56-
for r in range(accuracy)
55+
(-1) ** r * theta ** (2 * r + 1) / factorial(2 * r + 1) for r in range(accuracy)
5756
)
5857

5958

@@ -104,12 +103,14 @@ def maclaurin_cos(theta: float, accuracy: int = 30) -> float:
104103
theta = float(theta)
105104
div = theta // (2 * pi)
106105
theta -= 2 * div * pi
107-
return sum(
108-
(((-1) ** r) * ((theta ** (2 * r)) / factorial(2 * r))) for r in range(accuracy)
109-
)
106+
return sum((-1) ** r * theta ** (2 * r) / factorial(2 * r) for r in range(accuracy))
110107

111108

112109
if __name__ == "__main__":
110+
import doctest
111+
112+
doctest.testmod()
113+
113114
print(maclaurin_sin(10))
114115
print(maclaurin_sin(-10))
115116
print(maclaurin_sin(10, 15))

0 commit comments

Comments
 (0)