diff --git a/test/others.jl b/test/others.jl new file mode 100644 index 0000000..e9bef69 --- /dev/null +++ b/test/others.jl @@ -0,0 +1,17 @@ +@testset "Others" begin + @testset "Range" begin + elements = 10 + range_size = 5 + + df = DataFrame(x = [1:elements ...], y = [1:elements ...]) + conn = TablesConnector(df) + stream = BatchStream(conn) + + counter = 0 + for sample in Range(range_size, df) + counter += 1 + end + + @test counter == range_size + end +end \ No newline at end of file diff --git a/test/runtests.jl b/test/runtests.jl index c6c8fa1..db24ac9 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -3,4 +3,5 @@ using DataFrames using Test include("connector.jl") -include("stream.jl") \ No newline at end of file +include("stream.jl") +include("others.jl") \ No newline at end of file