diff --git a/safe_prime_generator_test.go b/safe_prime_generator_test.go index a0d1ba9..b4e58a5 100644 --- a/safe_prime_generator_test.go +++ b/safe_prime_generator_test.go @@ -2,7 +2,6 @@ package paillier import ( "errors" - "fmt" "math/big" "reflect" "testing" @@ -77,18 +76,18 @@ func TestAsyncGenerator(t *testing.T) { t.Errorf("2q + 1 != p") } - if len(p.Bytes()) != test.bitLen { - fmt.Errorf( + if p.BitLen() != test.bitLen { + t.Fatalf( "Unexpected p bit length\nActual: %v\nExpected: %v", - len(p.Bytes()), + p.BitLen(), test.bitLen, ) } - if len(q.Bytes()) != test.bitLen-1 { - fmt.Errorf( + if q.BitLen() != test.bitLen-1 { + t.Fatalf( "Unexpected q bit length\nActual: %v\nExpected: %v", - len(q.Bytes()), + q.BitLen(), test.bitLen-1, ) }