From 8c6419bde0cb6893cf7e789c8fceaea5638c95ff Mon Sep 17 00:00:00 2001
From: Adrien Hopkins <adrien.p.hopkins@gmail.com>
Date: Mon, 7 Aug 2023 15:24:54 -0500
Subject: Add tests for Prime Factorization function

---
 factors/prime_factorization.go | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'factors/prime_factorization.go')

diff --git a/factors/prime_factorization.go b/factors/prime_factorization.go
index 62498b5..3f3abc7 100644
--- a/factors/prime_factorization.go
+++ b/factors/prime_factorization.go
@@ -15,6 +15,10 @@ type PrimeFactorization struct {
 
 // PrimeFactorize creates a PrimeFactorization by factorizing a number.
 func PrimeFactorize(n uint) PrimeFactorization {
+	if n == 0 {
+		return PrimeFactorization{map[uint]uint{0: 1}}
+	}
+
 	unfactored := n // number with all found factors removed (divided)
 	exponents := make(map[uint]uint)
 
-- 
cgit v1.2.3