Browse Source

fix a bug while testing Quora Dataset

tags/v0.4.10
Yige Xu 5 years ago
parent
commit
d0a2b032bd
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      test/io/pipe/test_matching.py

+ 2
- 3
test/io/pipe/test_matching.py View File

@@ -80,9 +80,8 @@ class TestRunMatchingPipe(unittest.TestCase):
for k, v in data_set_dict.items():
path, pipe1, pipe2, data_set, vocab = v

with self.assertWarns(Warning):
data_bundle1 = pipe1(tokenizer='spacy').process_from_file(path)
data_bundle2 = pipe2(tokenizer='spacy').process_from_file(path)
data_bundle1 = pipe1(tokenizer='spacy').process_from_file(path)
data_bundle2 = pipe2(tokenizer='spacy').process_from_file(path)

self.assertTrue(isinstance(data_bundle1, DataBundle))
self.assertEqual(len(data_set), data_bundle1.num_dataset)


Loading…
Cancel
Save