From 051a0a80268a30d0b9e6aad7a9a42960a054688e Mon Sep 17 00:00:00 2001
From: Mr-Leshiy <leshiy12345678@gmail.com>
Date: Tue, 31 Oct 2023 12:03:22 +0200
Subject: [PATCH] add rounding to 1 decimal place

---
 .../src/proposal_score/mod.rs                 | 90 ++++++++++++++++++-
 1 file changed, 88 insertions(+), 2 deletions(-)

diff --git a/src/catalyst-toolbox/catalyst-toolbox/src/proposal_score/mod.rs b/src/catalyst-toolbox/catalyst-toolbox/src/proposal_score/mod.rs
index 6aff32305f..787a1aed2a 100644
--- a/src/catalyst-toolbox/catalyst-toolbox/src/proposal_score/mod.rs
+++ b/src/catalyst-toolbox/catalyst-toolbox/src/proposal_score/mod.rs
@@ -93,11 +93,13 @@ fn weighted_avarage_score(
     let allocated_weight = review_weight(allocated_weight, allocated_count);
     let not_allocated_weight = review_weight(not_allocated_weight, not_allocated_count);
 
-    let res = (total_allocated_rating as f64 * allocated_weight
+    let mut res = (total_allocated_rating as f64 * allocated_weight
         + total_not_allocated_rating as f64 * not_allocated_weight)
         / (allocated_weight * allocated_count as f64
             + not_allocated_weight * not_allocated_count as f64);
 
+    // round to 1 decimal place
+    res = (10.0 * res).round() / 10.0;
     Ok(res)
 }
 
@@ -119,7 +121,7 @@ mod tests {
     }
 
     #[test]
-    fn weighted_score_test() {
+    fn weighted_score_test_1() {
         let allocated_weight = 0.8;
         let not_allocated_weight = 0.2;
 
@@ -161,6 +163,90 @@ mod tests {
         assert!(weighted_avarage_score(0.5, 0.6, &[]).is_err());
     }
 
+    #[test]
+    fn weighted_score_test_2() {
+        let allocated_weight = 0.7;
+        let not_allocated_weight = 0.3;
+
+        let reviews = vec![
+            Review {
+                rating: 1,
+                allocated: false,
+            },
+            Review {
+                rating: 2,
+                allocated: false,
+            },
+            Review {
+                rating: 3,
+                allocated: false,
+            },
+            Review {
+                rating: 4,
+                allocated: false,
+            },
+            Review {
+                rating: 5,
+                allocated: false,
+            },
+            Review {
+                rating: 6,
+                allocated: true,
+            },
+            Review {
+                rating: 8,
+                allocated: true,
+            },
+        ];
+
+        let result =
+            weighted_avarage_score(allocated_weight, not_allocated_weight, &reviews).unwrap();
+        // To be precise the result should be `5.799999999999999`, but we are rounding to 1 decimal place
+        assert_eq!(result, 5.8);
+    }
+
+    #[test]
+    fn weighted_score_test_3() {
+        let allocated_weight = 0.7;
+        let not_allocated_weight = 0.3;
+
+        let reviews = vec![
+            Review {
+                rating: 1,
+                allocated: false,
+            },
+            Review {
+                rating: 2,
+                allocated: false,
+            },
+            Review {
+                rating: 3,
+                allocated: false,
+            },
+            Review {
+                rating: 4,
+                allocated: false,
+            },
+            Review {
+                rating: 5,
+                allocated: false,
+            },
+            Review {
+                rating: 6,
+                allocated: true,
+            },
+            Review {
+                rating: 7,
+                allocated: true,
+            },
+        ];
+
+        let result =
+            weighted_avarage_score(allocated_weight, not_allocated_weight, &reviews).unwrap();
+        // To be precise the result should be `5.449999999999999`, but we are rounding to 1 decimal place
+        assert_eq!(result, 5.4);
+    }
+
     #[test]
     fn full_test() {
         let allocated_weight = 0.8;