Separated the asserts in tests
@@ -28,7 +28,8 @@ def test_load_pretrained_vectors_word2vec():
assert isinstance(load_word2vec(dir_path), Word2VecKeyedVectors)
model = load_word2vec(dir_path, limit=500000)
- assert isinstance(model, Word2VecKeyedVectors) and (len(model.wv.vocab) == 500000)
+ assert isinstance(model, Word2VecKeyedVectors)
+ assert (len(model.wv.vocab) == 500000)
file_path = Path(file_path)
assert file_path.is_file()
@@ -45,7 +46,8 @@ def test_load_pretrained_vectors_glove():
assert isinstance(load_glove(dir_path), Word2VecKeyedVectors)
model = load_glove(dir_path, limit=50000)
- assert isinstance(model, Word2VecKeyedVectors) and (len(model.wv.vocab) == 50000)
+ assert isinstance(model, Word2VecKeyedVectors)
+ assert (len(model.wv.vocab) == 50000)
file_path = Path(file_path)
assert file_path.is_file()
