Update tests for data extraction

dan/datasets/extract/extract.py

@@ -281,7 +281,7 @@ class ArkindexExtractor:
         """
         return " ".join(
             [
-                self.mapping.encode[token] if token in self.mapping else token
+                self.mapping.encode[token] if token in self.mapping.encode else token
                 for token in list(text.strip())
             ]
         )