Typing

tests/test_segmentcollection.py

@@ -80,7 +80,7 @@ def test_get_segment_w_predicate():
 
 def test_split_by():
     def _serialize(collections: Iterable[RawSegmentCollection]) -> List[List[str]]:
-        lst = []
+        lst: list[str] = []
         global_lst = []
         for collection in collections:
             if lst:

tests/test_tokenizer.py

@@ -13,6 +13,7 @@
 #
 #    You should have received a copy of the GNU Lesser General Public License
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+from typing import Optional
 import pytest
 
 from pydifact.token import Token
@@ -22,12 +23,14 @@
 
 
 @pytest.fixture
-def tokenizer():
+def tokenizer() -> Tokenizer:
     return Tokenizer()
 
 
 def _assert_tokens(
-    collection: str, expected: list = None, error_message: str = None
+    collection: str,
+    expected: Optional[list] = None,
+    error_message: Optional[str] = None,
 ) -> None:
     """Helper function to accelerate tokenizer testing."""