kyber test

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
  "github.com/acheong08/crystals-go/dilithium"
+ "github.com/acheong08/crystals-go/kyber"
 )
 
 // Just testing a few things to make sure the code is working
@@ -15,4 +16,15 @@ func main() {
  } else {
  println("success")
  }
+ k := kyber.NewKyber512()
+ pk, sk = k.KeyGen([]byte("test"))
+ msg = []byte("hello world")
+ c, ss := k.Encaps(pk, msg)
+ newss := k.Decaps(sk, c)
+ // Convert to string and compare
+ if string(ss) != string(newss) {
+ panic("failed")
+ } else {
+ println("success")
+ }
 }