From ee7e96eaf6569f45845c13f5d5ee54f7ee8baa53 Mon Sep 17 00:00:00 2001
From: Piotr Dyraga <piotr.dyraga@akena.engineering>
Date: Fri, 27 Jul 2018 16:45:29 +0200
Subject: [PATCH] Fixed bit lenght assertions for safe prime generator tests

---
 safe_prime_generator_test.go | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

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,
 					)
 				}