#156 and #536 Unit-tests accelerated tests. Considering limited amoun…

…t of documents for iteration.
nicolay-r · Dec 27, 2023 · 0e4cc7f · 0e4cc7f
1 parent b76c236
commit 0e4cc7f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/contrib/source/test_nerel.py b/tests/contrib/source/test_nerel.py
@@ -30,8 +30,8 @@ def test(self):
  assert(isinstance(brat_relation, BratRelation))
  print(brat_relation.SourceID, brat_relation.TargetID, brat_relation.Type)
 
- def test_all_documents(self):
+ def test_all_documents(self, docs_limit=5):
  doc_reader = NerelDocReader(version=DEFAULT_VERSION)
- filenames_by_ids, folding = NerelIOUtils.read_dataset_split(version=DEFAULT_VERSION)
+ filenames_by_ids, folding = NerelIOUtils.read_dataset_split(version=DEFAULT_VERSION, docs_limit=docs_limit)
  for doc_id in tqdm(itertools.chain.from_iterable(folding.values())):
  doc_reader.read_document(filename=filenames_by_ids[doc_id], doc_id=0)
diff --git a/tests/contrib/source/test_nerel_bio.py b/tests/contrib/source/test_nerel_bio.py
@@ -30,8 +30,8 @@ def test(self):
  assert(isinstance(brat_relation, BratRelation))
  print(brat_relation.SourceID, brat_relation.TargetID, brat_relation.Type)
 
- def test_all_documents(self):
+ def test_all_documents(self, docs_limit=5):
  doc_reader = NerelBioDocReader(version=DEFAULT_VERSION)
- filenames_by_ids, folding = NerelBioIOUtils.read_dataset_split(version=DEFAULT_VERSION)
+ filenames_by_ids, folding = NerelBioIOUtils.read_dataset_split(version=DEFAULT_VERSION, docs_limit=docs_limit)
  for doc_id in tqdm(itertools.chain.from_iterable(folding.values())):
  doc_reader.read_document(filename=filenames_by_ids[doc_id], doc_id=0)