Spaces:
Build error
Build error
File size: 1,290 Bytes
dab2de2 077e1eb dab2de2 077e1eb dab2de2 077e1eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""Test files2df."""
from pathlib import Path
import tempfile
from radiobee.files2df import files2df
def test_files2df():
"""Test files2df with tests/test_en.txt tests/test_zh.txt."""
file1_ = "tests/test_en.txt"
file2_ = "tests/test_zh.txt"
with open(file1_, 'rb') as fh1, open(file2_, 'rb') as fh2:
file1 = tempfile._TemporaryFileWrapper(fh1, file1_)
file2 = tempfile._TemporaryFileWrapper(fh2, file2_)
assert Path(file1.name).is_file()
assert Path(file2.name).is_file()
df = files2df(file1, file2)
# with filenames as frist row
# assert df.iloc[1, 0] == "Wuthering Heights"
# assert df.iloc[1, 1] == "呼啸山庄"
assert df.iloc[0, 0] == "Wuthering Heights"
assert df.iloc[0, 1] == "呼啸山庄"
def test_files2df_file2none():
"""Test files2df with tests/test_en.txt None."""
file1_ = "tests/test_en.txt"
file2 = None
with open(file1_, 'rb') as fh1:
file1 = tempfile._TemporaryFileWrapper(fh1, file1_)
assert Path(file1.name).is_file()
df = files2df(file1, file2)
# with filename as first row
# assert df.iloc[1, 0] == "Wuthering Heights"
# assert df.iloc[1, 1] == ""
assert df.iloc[0, 0] == "Wuthering Heights"
assert df.iloc[0, 1] == ""
|