From c685548bccfac403cd938417a4650a04dd07800e Mon Sep 17 00:00:00 2001
From: n1net4il <n1net4il@kaist.ac.kr>
Date: Fri, 15 Dec 2023 12:17:15 +0000
Subject: [PATCH] improve test
 assignments::assignment10::small_exercises_grade::test_pythagorean

---
 .../assignment10/small_exercises_grade.rs         | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/assignments/assignment10/small_exercises_grade.rs b/src/assignments/assignment10/small_exercises_grade.rs
index 8cbc96f..595a708 100644
--- a/src/assignments/assignment10/small_exercises_grade.rs
+++ b/src/assignments/assignment10/small_exercises_grade.rs
@@ -340,12 +340,17 @@ mod test {
             (68, 285, 293),
         ];
 
-        for (i, t) in pythagorean().enumerate().take(1000) {
-            if i < pythagoreans.len() {
-                assert_eq!(pythagoreans[i], t)
-            }
-            let (a, b, c) = t;
+        let mut pythagorean_iter = pythagorean();
+        for i in 0..1000 {
+            let t = pythagorean_iter.next();
+            assert!(t.is_some());
+
+            let (a, b, c) = t.unwrap();
             assert_eq!(a * a + b * b, c * c);
+
+            if i < pythagoreans.len() {
+                assert_eq!(pythagoreans[i], (a, b, c))
+            }
         }
     }
 }