From 68d55463ababa6c4627526b86bdb43e5ee7c5a6b Mon Sep 17 00:00:00 2001
From: Denis Chaplygin <akashihi@gmail.com>
Date: Sun, 19 Dec 2021 18:19:46 +0200
Subject: [PATCH] Color specific test fixes

---
 src/epd5in83b_v2/graphics.rs | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/epd5in83b_v2/graphics.rs b/src/epd5in83b_v2/graphics.rs
index 92105f5..884fc8c 100644
--- a/src/epd5in83b_v2/graphics.rs
+++ b/src/epd5in83b_v2/graphics.rs
@@ -92,19 +92,25 @@ mod tests {
     #[test]
     fn graphics_size() {
         let display = Display5in83::default();
-        assert_eq!(display.buffer().len(), 38400);
+        assert_eq!(display.buffer().len(), 77760); // (77760 = 648 * 480/8) * 2
     }
 
     // test default background color on all bytes
     #[test]
     fn graphics_default() {
         let display = Display5in83::default();
-        for &byte in display.buffer() {
+        for &byte in display.bw_buffer() {
             assert_eq!(
                 byte,
                 epd5in83b_v2::DEFAULT_BACKGROUND_COLOR.get_byte_value()
             );
         }
+        for &byte in display.chromatic_buffer() {
+            assert_eq!(
+                byte,
+                0xFF
+            );
+        }
     }
 
     #[test]
@@ -114,7 +120,7 @@ mod tests {
             .into_styled(PrimitiveStyle::with_stroke(Black, 1))
             .draw(&mut display);
 
-        let buffer = display.buffer();
+        let buffer = display.bw_buffer();
 
         assert_eq!(buffer[0], Black.get_byte_value());
 
@@ -134,7 +140,7 @@ mod tests {
             .into_styled(PrimitiveStyle::with_stroke(Black, 1))
             .draw(&mut display);
 
-        let buffer = display.buffer();
+        let buffer = display.bw_buffer();
 
         assert_eq!(buffer[0], Black.get_byte_value());
 
@@ -154,7 +160,7 @@ mod tests {
             .into_styled(PrimitiveStyle::with_stroke(Black, 1))
             .draw(&mut display);
 
-        let buffer = display.buffer();
+        let buffer = display.bw_buffer();
 
         assert_eq!(buffer[0], Black.get_byte_value());
 
@@ -174,7 +180,7 @@ mod tests {
             .into_styled(PrimitiveStyle::with_stroke(Black, 1))
             .draw(&mut display);
 
-        let buffer = display.buffer();
+        let buffer = display.bw_buffer();
 
         assert_eq!(buffer[0], Black.get_byte_value());