Port tests

Author: Enkidu93

machine/corpora/__init__.py

@@ -16,6 +16,8 @@
 from .memory_stream_container import MemoryStreamContainer
 from .memory_text import MemoryText
 from .multi_key_ref import MultiKeyRef
+from .n_parallel_text_corpus import NParallelTextCorpus
+from .n_parallel_text_row import NParallelTextRow
 from .parallel_text_corpus import ParallelTextCorpus
 from .parallel_text_row import ParallelTextRow
 from .paratext_backup_terms_corpus import ParatextBackupTermsCorpus
@@ -121,6 +123,8 @@
     "nfkc_normalize",
     "nfkd_normalize",
     "normalize",
+    "NParallelTextCorpus",
+    "NParallelTextRow",
     "ParallelTextCorpus",
     "ParallelTextRow",
     "ParatextBackupTermsCorpus",

machine/corpora/n_parallel_text_corpus.py

@@ -112,7 +112,7 @@ def _get_text_ids_from_corpora(self) -> Set[str]:
         text_ids = text_ids.union(all_rows_text_ids)
         return text_ids
 
-    def get_rows(self, text_ids: Optional[Iterable[str]]) -> Iterable[NParallelTextRow]:
+    def get_rows(self, text_ids: Optional[Iterable[str]] = None) -> Iterable[NParallelTextRow]:
         filter_text_ids = self._get_text_ids_from_corpora()
         if text_ids is not None:
             filter_text_ids = filter_text_ids.intersection(text_ids)