From 7fb354fe43f10de0dc50cf7a60cc3978f3168074 Mon Sep 17 00:00:00 2001
From: Birk Tjelmeland <birktjelmeland@yahoo.no>
Date: Fri, 1 Feb 2019 11:19:21 +0100
Subject: [PATCH] Test decoder output

These tests are based on the assumtion that the current decoder is
correct and are to make sure that future changes to the decoder
wont break it.
---
 tests/decode_images.rs | 56 ++++++++++++++++++++++++++++++++++++------
 1 file changed, 48 insertions(+), 8 deletions(-)

diff --git a/tests/decode_images.rs b/tests/decode_images.rs
index 5881733..b654bf9 100644
--- a/tests/decode_images.rs
+++ b/tests/decode_images.rs
@@ -11,8 +11,18 @@ fn test_gray_u8()
     let img_file = File::open("./tests/images/minisblack-1c-8b.tiff").expect("Cannot find test image!");
     let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
     assert_eq!(decoder.colortype().unwrap(), ColorType::Gray(8));
-    let img_res = decoder.read_image();
-    assert!(img_res.is_ok());
+    let img_res = decoder.read_image().unwrap();
+    if let DecodingResult::U8(res) = img_res {
+        let mut res_sum: u64 = 0;
+        for x in res {
+            res_sum += x as u64;
+        }
+
+        assert_eq!(res_sum, 2840893);
+    }
+    else {
+        panic!("Wrong bit depth")
+    }
 }
 
 #[test]
@@ -21,8 +31,18 @@ fn test_rgb_u8()
     let img_file = File::open("./tests/images/rgb-3c-8b.tiff").expect("Cannot find test image!");
     let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
     assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(8));
-    let img_res = decoder.read_image();
-    assert!(img_res.is_ok());
+    let img_res = decoder.read_image().unwrap();
+    if let DecodingResult::U8(res) = img_res {
+        let mut res_sum: u64 = 0;
+        for x in res {
+            res_sum += x as u64;
+        }
+
+        assert_eq!(res_sum, 7842108)
+    }
+    else {
+        panic!("Wrong bit depth")
+    }
 }
 
 #[test]
@@ -31,8 +51,18 @@ fn test_gray_u16()
     let img_file = File::open("./tests/images/minisblack-1c-16b.tiff").expect("Cannot find test image!");
     let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
     assert_eq!(decoder.colortype().unwrap(), ColorType::Gray(16));
-    let img_res = decoder.read_image();
-    assert!(img_res.is_ok());
+    let img_res = decoder.read_image().unwrap();
+    if let DecodingResult::U16(res) = img_res {
+        let mut res_sum: u64 = 0;
+        for x in res {
+            res_sum += x as u64;
+        }
+
+        assert_eq!(res_sum, 733126239);
+    }
+    else {
+        panic!("Wrong bit depth")
+    }
 }
 
 #[test]
@@ -41,8 +71,18 @@ fn test_rgb_u16()
     let img_file = File::open("./tests/images/rgb-3c-16b.tiff").expect("Cannot find test image!");
     let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
     assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(16));
-    let img_res = decoder.read_image();
-    assert!(img_res.is_ok());
+    let img_res = decoder.read_image().unwrap();
+    if let DecodingResult::U16(res) = img_res {
+        let mut res_sum: u64 = 0;
+        for x in res {
+            res_sum += x as u64;
+        }
+
+        assert_eq!(res_sum, 2024349944);
+    }
+    else {
+        panic!("Wrong bit depth")
+    }
 }
 
 #[test]