Updated constants

root.go

@@ -3,14 +3,17 @@ package smt
 import "encoding/binary"
 
 const (
-	nonSumRootSizeBytes = 32
+	// These are intentionally exposed to allow for for testing and custom
+	// implementations of downstream applications.
+	SmtRootSizeBytes  = 32
+	SmstRootSizeBytes = SmtRootSizeBytes + sumSizeBytes + countSizeBytes
 )
 
 // Sum returns the uint64 sum of the merkle root, it checks the length of the
 // merkle root and if it is no the same as the size of the SMST's expected
 // root hash it will panic.
 func (r MerkleRoot) Sum() uint64 {
-	if len(r)%nonSumRootSizeBytes == 0 {
+	if len(r)%SmtRootSizeBytes == 0 {
 		panic("root#sum: not a merkle sum trie")
 	}
 
@@ -24,8 +27,8 @@ func (r MerkleRoot) Sum() uint64 {
 // Count returns the uint64 count of the merkle root, a cryptographically secure
 // count of the number of non-empty leafs in the tree.
 func (r MerkleRoot) Count() uint64 {
-	if len(r)%nonSumRootSizeBytes == 0 {
-		panic("root#count: not a merkle sum trie")
+	if len(r)%SmtRootSizeBytes == 0 {
+		panic("root#sum: not a merkle sum trie")
 	}
 
 	_, firstCountByteIdx := getFirstMetaByteIdx([]byte(r))