Skip to content

Commit

Permalink
Merge pull request #1218 from microsoft/bug/mind_smoke_test
Browse files Browse the repository at this point in the history
Bug/mind smoke test
  • Loading branch information
miguelgfierro authored Oct 21, 2020
2 parents 645a789 + c12bb8c commit 3402979
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/smoke/test_mind.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
def test_download_mind(tmp):
train_path, valid_path = download_mind(size="small", dest_path=tmp)
statinfo = os.stat(train_path)
assert statinfo.st_size == 54772957
assert statinfo.st_size == 52952752
statinfo = os.stat(valid_path)
assert statinfo.st_size == 32470157
assert statinfo.st_size == 30945572


@pytest.mark.smoke
Expand All @@ -21,19 +21,19 @@ def test_extract_mind(tmp):
train_path, valid_path = extract_mind(train_zip, valid_zip)

statinfo = os.stat(os.path.join(train_path, "behaviors.tsv"))
assert statinfo.st_size == 92047111
assert statinfo.st_size == 92019716
statinfo = os.stat(os.path.join(train_path, "entity_embedding.vec"))
assert statinfo.st_size == 25811015
statinfo = os.stat(os.path.join(train_path, "news.tsv"))
assert statinfo.st_size == 45895926
assert statinfo.st_size == 41202121
statinfo = os.stat(os.path.join(train_path, "relation_embedding.vec"))
assert statinfo.st_size == 1044588

statinfo = os.stat(os.path.join(valid_path, "behaviors.tsv"))
assert statinfo.st_size == 42975799
assert statinfo.st_size == 42838544
statinfo = os.stat(os.path.join(valid_path, "entity_embedding.vec"))
assert statinfo.st_size == 21960998
statinfo = os.stat(os.path.join(valid_path, "news.tsv"))
assert statinfo.st_size == 37410117
assert statinfo.st_size == 33519092
statinfo = os.stat(os.path.join(valid_path, "relation_embedding.vec"))
assert statinfo.st_size == 1044588

0 comments on commit 3402979

Please sign in to comment.