add unit test for countdigits

PeerDB-io · May 22, 2024 · e0c1207 · e0c1207
1 parent 98344e1
commit e0c1207
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/flow/datatypes/bigint_test.go b/flow/datatypes/bigint_test.go
@@ -0,0 +1,38 @@
+package datatypes
+
+import (
+	"math/big"
+	"testing"
+)
+
+func TestCountDigits(t *testing.T) {
+	bi := big.NewInt(1234567890)
+	expected := 10
+	result := CountDigits(bi)
+	if result != expected {
+		t.Errorf("Unexpected result. Expected: %v, but got: %v", expected, result)
+	}
+
+	bi = big.NewInt(-1234567890)
+	expected = 10
+	result = CountDigits(bi)
+	if result != expected {
+		t.Errorf("Unexpected result. Expected: %v, but got: %v", expected, result)
+	}
+
+	// 18 nines
+	bi = big.NewInt(999999999999999999)
+	result = CountDigits(bi)
+	expected = 18
+	if result != expected {
+		t.Errorf("Unexpected result. Expected: %v, but got: %v", expected, result)
+	}
+
+	// 18 nines
+	bi = big.NewInt(-999999999999999999)
+	result = CountDigits(bi)
+	expected = 18
+	if result != expected {
+		t.Errorf("Unexpected result. Expected: %v, but got: %v", expected, result)
+	}
+}