Use ruff for pytest style (#3872)

* Add ruff config for pytest

* Fixes for pre-commit

* Changes to make pre-commit happy (Unrelated to PR)

---------

Co-authored-by: Tristan Schulz <[email protected]>

Author: JasonGrace2282

manim/mobject/mobject.py

@@ -1988,7 +1988,8 @@ def restore(self) -> Self:
 
     def reduce_across_dimension(self, reduce_func: Callable, dim: int):
         """Find the min or max value from a dimension across all points in this and submobjects."""
-        assert dim >= 0 and dim <= 2
+        assert dim >= 0
+        assert dim <= 2
         if len(self.submobjects) == 0 and len(self.points) == 0:
             # If we have no points and no submobjects, return 0 (e.g. center)
             return 0

pyproject.toml

@@ -68,9 +68,7 @@ flake8-bugbear = "^23.11.28"
 flake8-builtins = "^2.2.0"
 flake8-comprehensions = "^3.7.0"
 flake8-docstrings = "^1.7.0"
-flake8-pytest-style = "^1.7.2"
 flake8-simplify = "^0.14.1"
-flake8-rst-docstrings = "^0.3.0"
 furo = "^2023.09.10"
 gitpython = "^3"
 isort = "^5.12.0"
@@ -139,13 +137,18 @@ select = [
   "E",
   "F",
   "I",
+  "PT",
   "UP",
 ]
 
 ignore = [
   # due to the import * used in manim
   "F403",
   "F405",
+  # fixtures not returning anything should have leading underscore
+  "PT004",
+  # Exception too broad (this would require lots of changes + re.escape) for little benefit
+  "PT011",
   # as recommended by https://docs.astral.sh/ruff/formatter/#conflicting-lint-rules
   "E111",
   "E114",
@@ -176,6 +179,10 @@ dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
 [tool.ruff.lint.isort]
 required-imports = ["from __future__ import annotations"]
 
+[tool.ruff.lint.flake8-pytest-style]
+fixture-parentheses = false
+mark-parentheses = false
+
 [tool.ruff.format]
 docstring-code-format = true
 

tests/module/animation/test_animate.py

@@ -23,10 +23,8 @@ def test_chained_animate():
     scale_factor = 2
     direction = np.array((1, 1, 0))
     anim = s.animate.scale(scale_factor).shift(direction).build()
-    assert (
-        anim.mobject.target.width == scale_factor * s.width
-        and (anim.mobject.target.get_center() == direction).all()
-    )
+    assert anim.mobject.target.width == scale_factor * s.width
+    assert (anim.mobject.target.get_center() == direction).all()
 
 
 def test_overridden_animate():
@@ -103,10 +101,8 @@ def test_chained_animate_with_args():
     run_time = 2
 
     anim = s.animate(run_time=run_time).scale(scale_factor).shift(direction).build()
-    assert (
-        anim.mobject.target.width == scale_factor * s.width
-        and (anim.mobject.target.get_center() == direction).all()
-    )
+    assert anim.mobject.target.width == scale_factor * s.width
+    assert (anim.mobject.target.get_center() == direction).all()
     assert anim.run_time == run_time
 
 

tests/module/animation/test_composition.py

@@ -132,7 +132,7 @@ def test_animationgroup_with_wait():
 
 
 @pytest.mark.parametrize(
-    "animation_remover, animation_group_remover",
+    ("animation_remover", "animation_group_remover"),
     [(False, True), (True, False)],
 )
 def test_animationgroup_is_passing_remover_to_animations(

tests/module/mobject/graphing/test_coordinate_system.py

@@ -56,7 +56,7 @@ def test_dimension():
 
 def test_abstract_base_class():
     """Check that CoordinateSystem has some abstract methods."""
-    with pytest.raises(Exception):
+    with pytest.raises(NotImplementedError):
         CS().get_axes()
 
 

tests/module/mobject/test_boolean_ops.py

@@ -8,7 +8,7 @@
 
 
 @pytest.mark.parametrize(
-    "test_input,expected",
+    ("test_input", "expected"),
     [
         (
             [(1.0, 2.0), (3.0, 4.0)],

tests/module/mobject/text/test_texmobject.py

@@ -19,7 +19,7 @@ def test_SingleStringMathTex(config):
 
 
 @pytest.mark.parametrize(  # : PT006
-    "text_input,length_sub",
+    ("text_input", "length_sub"),
     [("{{ a }} + {{ b }} = {{ c }}", 5), (r"\frac{1}{a+b\sqrt{2}}", 1)],
 )
 def test_double_braces_testing(text_input, length_sub):

tests/module/scene/test_auto_zoom.py

@@ -16,6 +16,7 @@ def test_zoom():
 
     assert scene.camera.frame_width == abs(
         s1.get_left()[0] - s2.get_right()[0],
-    ) and scene.camera.frame.get_center()[0] == (
+    )
+    assert scene.camera.frame.get_center()[0] == (
         abs(s1.get_center()[0] + s2.get_center()[0]) / 2
     )

tests/opengl/test_animate_opengl.py

@@ -23,10 +23,8 @@ def test_chained_animate(using_opengl_renderer):
     scale_factor = 2
     direction = np.array((1, 1, 0))
     anim = s.animate.scale(scale_factor).shift(direction).build()
-    assert (
-        anim.mobject.target.width == scale_factor * s.width
-        and (anim.mobject.target.get_center() == direction).all()
-    )
+    assert anim.mobject.target.width == scale_factor * s.width
+    assert (anim.mobject.target.get_center() == direction).all()
 
 
 def test_overridden_animate(using_opengl_renderer):
@@ -103,10 +101,8 @@ def test_chained_animate_with_args(using_opengl_renderer):
     run_time = 2
 
     anim = s.animate(run_time=run_time).scale(scale_factor).shift(direction).build()
-    assert (
-        anim.mobject.target.width == scale_factor * s.width
-        and (anim.mobject.target.get_center() == direction).all()
-    )
+    assert anim.mobject.target.width == scale_factor * s.width
+    assert (anim.mobject.target.get_center() == direction).all()
     assert anim.run_time == run_time
 
 

tests/opengl/test_coordinate_system_opengl.py

@@ -55,7 +55,7 @@ def test_dimension(using_opengl_renderer):
 
 def test_abstract_base_class(using_opengl_renderer):
     """Check that CoordinateSystem has some abstract methods."""
-    with pytest.raises(Exception):
+    with pytest.raises(NotImplementedError):
         CS().get_axes()