Minor type changes

automata/base/automaton.py

@@ -151,6 +151,10 @@ def __repr__(self) -> str:
         )
         return f"{self.__class__.__qualname__}({values})"
 
-    def __contains__(self, input_str: str) -> bool:
+    def __contains__(self, item: Any) -> bool:
         """Returns whether the word is accepted by the automaton."""
-        return self.accepts_input(input_str)
+
+        if not isinstance(item, str):
+            return False
+
+        return self.accepts_input(item)

tests/test_dfa.py

@@ -178,6 +178,7 @@ def test_accepts_input_false(self) -> None:
         """Should return False if DFA input is rejected."""
         self.assertFalse(self.dfa.accepts_input("011"))
         self.assertNotIn("011", self.dfa)
+        self.assertNotIn(1, self.nfa)
 
     def test_read_input_step(self) -> None:
         """Should return validation generator if step flag is supplied."""

tests/test_nfa.py

@@ -233,6 +233,7 @@ def test_accepts_input_false(self) -> None:
         """Should return False if NFA input is rejected."""
         self.assertFalse(self.nfa.accepts_input("abba"))
         self.assertNotIn("abba", self.nfa)
+        self.assertNotIn(1, self.nfa)
 
     def test_cyclic_lambda_transitions(self) -> None:
         """Should traverse NFA containing cyclic lambda transitions."""