From 7a32f565f595e23dc6865258c4e61476c926992c Mon Sep 17 00:00:00 2001
From: Marek <mail@marek.onl>
Date: Wed, 8 Nov 2023 15:39:16 +0100
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: teor <teor@riseup.net>
---
 zebra-chain/src/parallel/tree.rs | 2 +-
 zebra-state/src/request.rs       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/zebra-chain/src/parallel/tree.rs b/zebra-chain/src/parallel/tree.rs
index cd751765838..98df543735c 100644
--- a/zebra-chain/src/parallel/tree.rs
+++ b/zebra-chain/src/parallel/tree.rs
@@ -11,7 +11,7 @@ use crate::{
 };
 
 /// An argument wrapper struct for note commitment trees.
-#[derive(Clone, Debug, Default)]
+#[derive(Clone, Debug, Default, Eq, PartialEq)]
 pub struct NoteCommitmentTrees {
     /// The sprout note commitment tree.
     pub sprout: Arc<sprout::tree::NoteCommitmentTree>,
diff --git a/zebra-state/src/request.rs b/zebra-state/src/request.rs
index 027b2f8c6bb..ec270bb7a0a 100644
--- a/zebra-state/src/request.rs
+++ b/zebra-state/src/request.rs
@@ -224,7 +224,7 @@ pub struct ContextuallyVerifiedBlock {
 }
 
 /// Wraps note commitment trees and the history tree together.
-#[derive(Default)]
+#[derive(Clone, Debug, Default, Eq, PartialEq)]
 pub struct Treestate {
     /// Note commitment trees.
     pub note_commitment_trees: NoteCommitmentTrees,