🎨 Migrate to Pytest

pyformlang/cfg/cfg.py

@@ -1,7 +1,7 @@
 """ A context free grammar """
 import string
 from copy import deepcopy
-from typing import AbstractSet, Iterable, Tuple, Dict, Any
+from typing import AbstractSet, Iterable, Tuple, Dict, Any, Union
 
 import networkx as nx
 
@@ -56,9 +56,9 @@ class CFG:
     # pylint: disable=too-many-instance-attributes
 
     def __init__(self,
-                 variables: AbstractSet[Variable] = None,
-                 terminals: AbstractSet[Terminal] = None,
-                 start_symbol: Variable = None,
+                 variables: AbstractSet[Union[Variable, str]] = None,
+                 terminals: AbstractSet[Union[Terminal, str]] = None,
+                 start_symbol: Union[Variable, str] = None,
                  productions: Iterable[Production] = None):
         if variables is not None:
             variables = {to_variable(x) for x in variables}
@@ -706,10 +706,10 @@ def is_empty(self) -> bool:
     def __bool__(self):
         return not self.is_empty()
 
-    def __contains__(self, word: Iterable[Terminal]) -> bool:
+    def __contains__(self, word: Iterable[Union[Terminal, str]]) -> bool:
         return self.contains(word)
 
-    def contains(self, word: Iterable[Terminal]) -> bool:
+    def contains(self, word: Iterable[Union[Terminal, str]]) -> bool:
         """ Gives the membership of a word to the grammar
 
         Parameters

pyformlang/cfg/cyk_table.py

@@ -12,7 +12,7 @@ class CYKTable:
     Parameters
     ----------
     cfg : A context-free grammar
-    word : tuple of Terminals
+    word : iterable of Terminals
         The word from which we construct the CYK table
     """
 

pyformlang/cfg/terminal.py

@@ -4,13 +4,7 @@
 
 
 class Terminal(CFGObject):  # pylint: disable=too-few-public-methods
-    """ An terminal in a CFG
-
-    Parameters
-    -----------
-    value : any
-        The value of the terminal
-    """
+    """ A terminal in a CFG """
 
     def __eq__(self, other):
         return isinstance(other, Terminal) and self.value == other.value