add detect_compressor() tests

JuliaIO · Mar 17, 2021 · 50e9cbf · 50e9cbf
1 parent 4d4fd70
commit 50e9cbf
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 0 deletions.
diff --git a/test/files/dummy.txt b/test/files/dummy.txt
@@ -0,0 +1 @@
+This is a dummy file for FileIO.detect_compressor() test
diff --git a/test/files/dummy.txt.bz2 b/test/files/dummy.txt.bz2
diff --git a/test/files/dummy.txt.gz b/test/files/dummy.txt.gz
diff --git a/test/files/dummy.txt.lz4 b/test/files/dummy.txt.lz4
diff --git a/test/files/dummy.txt.xz b/test/files/dummy.txt.xz
diff --git a/test/query.jl b/test/query.jl
@@ -318,6 +318,26 @@ end
     end
 end
 
+@testset "detect $format compression" for (ext, format) in [(nothing, nothing), (".gz", "GZIP"), (".bz2", "BZIP2"),
+                                                            (".lz4", "LZ4"), (".xz", "XZ")]
+    fname = joinpath(@__DIR__, "files", "dummy.txt")
+    if ext !== nothing
+        fname *= ext
+    end
+    open(fname) do io
+        @test FileIO.detect_compressor(io, formats=[format]) == format # test with specific format only
+    end
+    open(fname) do io
+        @test FileIO.detect_compressor(io, formats=[]) === nothing # test with no formats
+    end
+    open(fname) do io
+        @test FileIO.detect_compressor(io) == format # test with all formats
+    end
+    open(fname) do io
+        @test FileIO.detect_compressed(io) == (format !== nothing)
+    end
+end
+
 let file_dir = joinpath(@__DIR__, "files"), file_path = Path(file_dir)
     @testset "Querying with $(typeof(fp))" for fp in (file_dir, file_path)
         @testset "bedGraph" begin