make stream types a subtype of TranscodingStream

JuliaIO · Jun 11, 2017 · dc697b1 · dc697b1
1 parent f4bb55d
commit dc697b1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/compression.jl b/src/compression.jl
@@ -36,7 +36,7 @@ function Bzip2Compression(;blocksize100k::Integer=DEFAULT_BLOCKSIZE100K,
     return Bzip2Compression(BZStream(), blocksize100k, workfactor, verbosity)
 end
 
-const Bzip2CompressionStream{S} = TranscodingStream{Bzip2Compression,S}
+const Bzip2CompressionStream{S} = TranscodingStream{Bzip2Compression,S} where S<:IO
 
 """
     Bzip2CompressionStream(stream::IO; kwargs...)

diff --git a/src/decompression.jl b/src/decompression.jl
@@ -24,7 +24,7 @@ function Bzip2Decompression(;small::Bool=false, verbosity::Integer=DEFAULT_VERBO
     return Bzip2Decompression(BZStream(), small, verbosity)
 end
 
-const Bzip2DecompressionStream{S} = TranscodingStream{Bzip2Decompression,S}
+const Bzip2DecompressionStream{S} = TranscodingStream{Bzip2Decompression,S} where S<:IO
 
 """
     Bzip2DecompressionStream(stream::IO; kwargs...)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -15,6 +15,9 @@ import TranscodingStreams
     write(stream, b"bar", TranscodingStreams.TOKEN_END)
     @test read(Bzip2DecompressionStream(IOBuffer(take!(buf)))) == b"foobar"
 
+    @test Bzip2CompressionStream <: TranscodingStreams.TranscodingStream
+    @test Bzip2DecompressionStream <: TranscodingStreams.TranscodingStream
+
     TranscodingStreams.test_roundtrip_read(Bzip2CompressionStream, Bzip2DecompressionStream)
     TranscodingStreams.test_roundtrip_write(Bzip2CompressionStream, Bzip2DecompressionStream)
     TranscodingStreams.test_roundtrip_lines(Bzip2CompressionStream, Bzip2DecompressionStream)