(tmpdir, compression_opener)
| 295 | |
| 296 | @pytest.mark.parametrize("compression_opener", [(None, open), ("gzip", gzip.open)]) |
| 297 | def test_open_files_write(tmpdir, compression_opener): |
| 298 | compression, opener = compression_opener |
| 299 | tmpdir = str(tmpdir) |
| 300 | files = open_files(tmpdir, num=2, mode="wb", compression=compression) |
| 301 | assert len(files) == 2 |
| 302 | assert {f.mode for f in files} == {"wb"} |
| 303 | for fil in files: |
| 304 | with fil as f: |
| 305 | f.write(b"000") |
| 306 | files = sorted(os.listdir(tmpdir)) |
| 307 | assert files == ["0.part", "1.part"] |
| 308 | |
| 309 | with opener(os.path.join(tmpdir, files[0]), "rb") as f: |
| 310 | d = f.read() |
| 311 | assert d == b"000" |
| 312 | |
| 313 | |
| 314 | def test_pickability_of_lazy_files(tmpdir): |
nothing calls this directly
no test coverage detected
searching dependent graphs…