tests for PySet

Author: Christopher Doris

.gitignore

@@ -5,3 +5,4 @@ __pycache__
 build/
 dist/
 .CondaPkg/
+/jltest.*

test/convert.jl

@@ -169,6 +169,15 @@ end
     @test t4 === (x=6, y=7)
 end
 
+@testitem "mapping → PyDict" begin
+    x1 = pyconvert(PyDict, pydict([1=>11, 2=>22, 3=>33]))
+    @test x1 isa PyDict{Py,Py}
+    @test isequal(x1, Dict([Py(1)=>Py(11), Py(2)=>Py(22), Py(3)=>Py(33)]))
+    x2 = pyconvert(PyDict{Int,Int}, pydict([1=>11, 2=>22, 3=>33]))
+    @test x2 isa PyDict{Int,Int}
+    @test x2 == Dict(1=>11, 2=>22, 3=>33)
+end
+
 @testitem "mapping → Dict" begin
     x1 = pyconvert(Dict, pydict(["a"=>1, "b"=>2]))
     @test x1 isa Dict{String, Int}
@@ -178,6 +187,24 @@ end
     @test x2 == Dict('c'=>3.0, 'd'=>4.0)
 end
 
+@testitem "sequence → PyList" begin
+    x1 = pyconvert(PyList, pylist([1, 2, 3]))
+    @test x1 isa PyList{Py}
+    @test isequal(x1, [Py(1), Py(2), Py(3)])
+    x2 = pyconvert(PyList{Int}, pylist([1, 2, 3]))
+    @test x2 isa PyList{Int}
+    @test x2 == [1, 2, 3]
+end
+
+@testitem "set → PySet" begin
+    x1 = pyconvert(PySet, pyset([1, 2, 3]))
+    @test x1 isa PySet{Py}
+    @test isequal(x1, Set([Py(1), Py(2), Py(3)]))
+    x2 = pyconvert(PySet{Int}, pyset([1, 2, 3]))
+    @test x2 isa PySet{Int}
+    @test x2 == Set([1, 2, 3])
+end
+
 @testitem "date → Date" begin
     using Dates
     x1 = pyconvert(Date, pydate(2001, 2, 3))

test/pywrap.jl

@@ -272,13 +272,101 @@ end
 end
 
 @testitem "PySet" begin
+    x = pyset([1, 2, 3])
+    y = PySet(x)
+    z = PySet{Int}(x)
+    e = PySet{String}()
+    @testset "construct" begin
+        @test y isa PySet{Py}
+        @test z isa PySet{Int}
+        @test e isa PySet{String}
+        @test PythonCall.ispy(y)
+        @test PythonCall.ispy(z)
+        @test PythonCall.ispy(e)
+        @test Py(y) === x
+        @test Py(z) === x
+        @test Py(e) !== x
+    end
+    @testset "length" begin
+        @test length(y) == 3
+        @test length(z) == 3
+        @test length(e) == 0
+    end
+    @testset "isempty" begin
+        @test !isempty(y)
+        @test !isempty(z)
+        @test isempty(e)
+    end
+    @testset "in" begin
+        @test 1 in z
+        @test 3 in z
+        @test !(4 in z)
+        @test !(1 in e)
+        @test 2.0 in z
+        @test !(2.1 in z)
+    end
+    @testset "push!" begin
+        a = PySet{String}(["a"])
+        @test a == Set(["a"])
+        push!(a, "b")
+        @test a == Set(["a", "b"])
+        push!(a, "a")
+        @test a == Set(["a", "b"])
+    end
+    @testset "delete!" begin
+        a = PySet{Int}()
+        @test a == Set()
+        delete!(a, 0)
+        @test a == Set()
+        delete!(a, nothing)
+        @test a == Set()
+        push!(a, 1, 2, 3)
+        @test a == Set([1, 2, 3])
+        delete!(a, 2)
+        @test a == Set([1, 3])
+        delete!(a, 1.2)
+        @test a == Set([1, 3])
+        delete!(a, 3.0)
+        @test a == Set([1])
+        delete!(a, nothing)
+        @test a == Set([1])
+    end
+    @testset "pop!" begin
+        a = PySet{Int}()
+        @test_throws Exception pop!(a)
+        push!(a, 1, -1)
+        x = pop!(a)
+        @test x in [1, -1]
+        @test a == Set([-x])
+        y = pop!(a)
+        @test x == -y
+        @test a == Set()
+        @test_throws Exception pop!(a)
+        push!(a, 1, 2, 3)
+        @test pop!(a, 1) == 1
+        @test a == Set([2, 3])
+        @test pop!(a, 2.0) == 2
+        @test a == Set([3])
+        @test_throws Exception pop!(a, 1)
+        @test_throws Exception pop!(a, 1.0)
+        @test_throws Exception pop!(a, nothing)
+        @test a == Set([3])
+        @test pop!(a, 1, 99) === 99
+        @test pop!(a, 1.0, 99) === 99
+        @test pop!(a, nothing, 99) === 99
+    end
+    @testset "empty!" begin
+        a = PySet{Int}([1, 2, 3])
+        @test a == Set([1, 2, 3])
+        @test empty!(a) === a
+        @test a == Set()
+    end
     @testset "copy" begin
-        x = PySet{Int}([1,2,3])
-        y = copy(x)
-        @test y isa PySet{Int}
-        push!(y, 99)
-        @test x == Set([1, 2, 3])
-        @test y == Set([1, 2, 3, 99])
+        z2 = copy(z)
+        @test z2 isa PySet{Int}
+        push!(z2, 99)
+        @test z == Set([1, 2, 3])
+        @test z2 == Set([1, 2, 3, 99])
     end
 end