format tests with ruff

FormalLanguageConstrainedPathQuerying · Jan 3, 2025 · d006c35 · d006c35
1 parent 148445f
commit d006c35
Show file tree

Hide file tree

Showing 24 changed files with 1,754 additions and 1,388 deletions.
diff --git a/pyformlang/cfg/tests/test_cfg.py b/pyformlang/cfg/tests/test_cfg.py
diff --git a/pyformlang/cfg/tests/test_llone_parser.py b/pyformlang/cfg/tests/test_llone_parser.py
diff --git a/pyformlang/cfg/tests/test_production.py b/pyformlang/cfg/tests/test_production.py
@@ -1,13 +1,14 @@
-""" Tests the productions """
+"""Tests the productions."""
 
 from pyformlang.cfg import Production, Variable, Terminal
 
 
 class TestProduction:
-    """ Tests the production """
+    """Tests the production."""
+
     # pylint: disable=missing-function-docstring
 
-    def test_creation(self):
+    def test_creation(self) -> None:
         prod0 = Production(Variable("S0"), [Terminal("S1"), Variable("a")])
         prod1 = Production(Variable("S0"), [Terminal("S1"), Variable("a")])
         prod2 = Production(Variable("S0'"), [Terminal("S1"), Variable("a")])

diff --git a/pyformlang/cfg/tests/test_recursive_decent_parser.py b/pyformlang/cfg/tests/test_recursive_decent_parser.py
@@ -2,8 +2,10 @@
 # pylint: disable=missing-class-docstring
 # pylint: disable=missing-function-docstring
 from pyformlang.cfg import CFG, Variable, Terminal
-from pyformlang.cfg.recursive_decent_parser import \
-    RecursiveDecentParser, NotParsableError
+from pyformlang.cfg.recursive_decent_parser import (
+    RecursiveDecentParser,
+    NotParsableError,
+)
 import pytest
 
 
@@ -19,43 +21,84 @@ def parser():
 
 
 class TestRecursiveDecentParser:
-
-    def test_creation(self, parser):
+    def test_creation(self, parser) -> None:
         assert parser is not None
 
-    def test_get_parsing_tree(self, parser):
+    def test_get_parsing_tree(self, parser) -> None:
         assert parser.is_parsable(
             ["(", "int", "+", "(", "int", "*", "int", ")", ")"]
         )
         parse_tree = parser.get_parse_tree(
-            ["(", "int", "+", "(", "int", "*", "int", ")", ")"])
+            ["(", "int", "+", "(", "int", "*", "int", ")", ")"]
+        )
         derivation = parse_tree.get_leftmost_derivation()
-        assert derivation == \
-            [[Variable("S")],
-             [Terminal("("), Variable("E"), Terminal(")")],
-             [Terminal("("), Variable("S"), Terminal("+"), Variable("S"),
-              Terminal(")")],
-             [Terminal("("), Terminal("int"), Terminal("+"), Variable("S"),
-              Terminal(")")],
-             [Terminal("("), Terminal("int"), Terminal("+"), Terminal("("),
-              Variable("E"), Terminal(")"), Terminal(")")],
-             [Terminal("("), Terminal("int"), Terminal("+"), Terminal("("),
-              Variable("S"), Terminal("*"), Variable("S"), Terminal(")"),
-              Terminal(")")],
-             [Terminal("("), Terminal("int"), Terminal("+"), Terminal("("),
-              Terminal("int"), Terminal("*"), Variable("S"), Terminal(")"),
-              Terminal(")")],
-             [Terminal("("), Terminal("int"), Terminal("+"), Terminal("("),
-              Terminal("int"), Terminal("*"), Terminal("int"), Terminal(")"),
-              Terminal(")")],
-             ]
+        assert derivation == [
+            [Variable("S")],
+            [Terminal("("), Variable("E"), Terminal(")")],
+            [
+                Terminal("("),
+                Variable("S"),
+                Terminal("+"),
+                Variable("S"),
+                Terminal(")"),
+            ],
+            [
+                Terminal("("),
+                Terminal("int"),
+                Terminal("+"),
+                Variable("S"),
+                Terminal(")"),
+            ],
+            [
+                Terminal("("),
+                Terminal("int"),
+                Terminal("+"),
+                Terminal("("),
+                Variable("E"),
+                Terminal(")"),
+                Terminal(")"),
+            ],
+            [
+                Terminal("("),
+                Terminal("int"),
+                Terminal("+"),
+                Terminal("("),
+                Variable("S"),
+                Terminal("*"),
+                Variable("S"),
+                Terminal(")"),
+                Terminal(")"),
+            ],
+            [
+                Terminal("("),
+                Terminal("int"),
+                Terminal("+"),
+                Terminal("("),
+                Terminal("int"),
+                Terminal("*"),
+                Variable("S"),
+                Terminal(")"),
+                Terminal(")"),
+            ],
+            [
+                Terminal("("),
+                Terminal("int"),
+                Terminal("+"),
+                Terminal("("),
+                Terminal("int"),
+                Terminal("*"),
+                Terminal("int"),
+                Terminal(")"),
+                Terminal(")"),
+            ],
+        ]
 
-    def test_no_parse_tree(self, parser):
+    def test_no_parse_tree(self, parser) -> None:
         with pytest.raises(NotParsableError):
             parser.get_parse_tree([")"])
         assert not (parser.is_parsable([")"]))
 
-    def test_infinite_recursion(self):
+    def test_infinite_recursion(self) -> None:
         cfg = CFG.from_text("""
             S -> S E
         """)

diff --git a/pyformlang/cfg/tests/test_terminal.py b/pyformlang/cfg/tests/test_terminal.py
@@ -1,13 +1,15 @@
-""" Tests the terminal """
+"""Tests the terminal."""
+
 from pyformlang.cfg import Variable, Terminal, Epsilon
 from pyformlang.finite_automaton import State, Symbol, Epsilon as FAEpsilon
 
 
 class TestTerminal:
-    """ Tests the terminal """
+    """Tests the terminal."""
+
     # pylint: disable=missing-function-docstring
 
-    def test_creation(self):
+    def test_creation(self) -> None:
         terminal0 = Terminal(0)
         terminal1 = Terminal(1)
         terminal2 = Terminal(0)
@@ -27,7 +29,7 @@ def test_creation(self):
         assert str(terminal0) == "0"
         assert repr(terminal0) == "Terminal(0)"
 
-    def test_eq(self):
+    def test_eq(self) -> None:
         assert "epsilon" == Epsilon()
         assert Epsilon() == "ɛ"
         assert Terminal("A") != Variable("A")

diff --git a/pyformlang/cfg/tests/test_variable.py b/pyformlang/cfg/tests/test_variable.py
@@ -1,12 +1,14 @@
-""" Tests the variable """
+"""Tests the variable."""
+
 from pyformlang.cfg import Variable
 
 
 class TestVariable:
-    """ Tests the variable """
+    """Tests the variable."""
+
     # pylint: disable=missing-function-docstring
 
-    def test_creation(self):
+    def test_creation(self) -> None:
         variable0 = Variable(0)
         variable1 = Variable(1)
         variable2 = Variable(0)