add tests

Author: nzy1997

docs/src/index.md

@@ -7,7 +7,6 @@ A repository for *quantum error correction* (QEC) codes.
 
 
 ## Quantum Code list
-
 | Code name    | Struct |
 | -------- | ------- |
 | Perfect 5-qubit code | [`Perfect5`](@ref) |

src/codes/classical/repetition.jl

@@ -20,12 +20,12 @@ Bitflip3 is a quantum error correction code that corrects single-qubit bit-flip
 """
 struct Bitflip3 <: AbstractQECC end
 
-parity_matrix(c::Bitflip3) = hcat(zeros(Bool,3,3), parity_matrix(RepCode(3)))
+parity_matrix(c::Bitflip3) = hcat(zeros(Bool,2,3), parity_matrix(RepCode(3))[1:2,:])
 
 """Three-qubit phase-flip code.
 
 Phaseflip3 is a quantum error correction code that corrects single-qubit phase-flip error and does not detect any bit-flip errors.
 """
 struct Phaseflip3 <: AbstractQECC end
 
-parity_matrix(c::Phaseflip3) = hcat(parity_matrix(RepCode(3)), zeros(Bool,3,3))
+parity_matrix(c::Phaseflip3) = hcat(parity_matrix(RepCode(3))[1:2,:], zeros(Bool,2,3))

test/codes/classical/repetition.jl

@@ -10,4 +10,16 @@ using QECC
                                   1 0 0 0 1]
     @test code_n(c) == 5
     @test code_s(c) == 5
+end
+
+@testset "Bitflip3" begin
+    c = Bitflip3()
+    @test parity_matrix(c) == Bool[0 0 0 1 1 0;
+                                  0 0 0 0 1 1]
+end
+
+@testset "Phaseflip3" begin
+    c = Phaseflip3()
+    @test parity_matrix(c) == Bool[1 1 0 0 0 0;
+                                  0 1 1 0 0 0]
 end

test/codes/css.jl

@@ -0,0 +1,10 @@
+using Test
+using QECC
+
+@testset "CSS" begin
+    h = QECC._steane_mat()
+    c = CSS(h, h)
+    @test parity_matrix(c) == parity_matrix(Steane7())
+    @test parity_matrix_x(c) == h
+    @test parity_matrix_z(c) == h
+end

test/codes/surfacecode.jl

@@ -7,6 +7,10 @@ using QECC
                                   0 1 0 1 1 0 0 0 0 0;
                                   0 0 0 0 0 1 1 0 0 1;
                                   0 0 0 0 0 0 0 1 1 1]
+    @test parity_matrix_x(c) == Bool[1 0 1 0 1;
+                                    0 1 0 1 1]
+    @test parity_matrix_z(c) == Bool[1 1 0 0 1;
+                                    0 0 1 1 1]
     @test distance(c) == 2
     @test code_n(c) == 5
     @test code_s(c) == 4

test/codes/toric.jl

@@ -3,12 +3,15 @@ using QECC
 
 @testset "Toric" begin
     c = Toric(2,2)
-    @test parity_matrix(c) == Bool[1  0  1  0  1  1  0  0  0  0  0  0  0  0  0  0;
+    pm = Bool[1  0  1  0  1  1  0  0  0  0  0  0  0  0  0  0;
     0  1  0  1  1  1  0  0  0  0  0  0  0  0  0  0;
     1  0  1  0  0  0  1  1  0  0  0  0  0  0  0  0;
     0  0  0  0  0  0  0  0  1  1  0  0  1  0  1  0;
     0  0  0  0  0  0  0  0  1  1  0  0  0  1  0  1;
     0  0  0  0  0  0  0  0  0  0  1  1  1  0  1  0]
+    @test parity_matrix(c) == pm
+    @test parity_matrix_x(c) == pm[1:3,1:8]
+    @test parity_matrix_z(c) == pm[4:6,9:end]
     @test distance(c) == 2
     @test code_n(c) == 8
     @test code_s(c) == 6

test/runtests.jl

@@ -27,4 +27,8 @@ end
 
 @testset "clevecode" begin
     include("codes/clevecode.jl")
+end
+
+@testset "css" begin
+    include("codes/css.jl")
 end