test saturation on the undistorted image to be independent of cv2 , print from saturation test removed

Felix Darvas · Felix Darvas · commit bb937cc0a2d0 · 2020-11-06T14:43:57.000-08:00
diff --git a/tests/test_panel.py b/tests/test_panel.py
@@ -87,16 +87,15 @@ def test_raw_panel_saturatedl(panel_image_name):
     
     #saturate 2500 pixels in the raw image - note that on the undistorted image this
     #will result in 2329 saturated pixels
-    i0 = img.raw()
+    i0 = img.undistorted(img.raw())
     i0[500:550,700:750]=4095*16+1
-    img.set_raw(i0)
+    img.set_undistorted(i0)
     
     mean, std, num, sat = pan.raw()
-    print(mean,std,num,sat)
     assert mean == pytest.approx(47245, rel=0.01)
     assert std == pytest.approx(5846.1, rel=0.05)
     assert num == pytest.approx(26005, rel=0.001)
-    assert sat == pytest.approx(2329, abs=0)
+    assert sat == pytest.approx(2500, abs=0)
 
 def test_raw_panel(panel_image_name):
     img = image.Image(panel_image_name)
