@@ -102,16 +102,16 @@ def _show_diff_helper(self, frame_data, expected_frame_data):
102
102
diff_im = expected_frame_data .copy ()
103
103
diff_im = np .where (
104
104
frame_data != np .array ([0 , 0 , 0 , 255 ]),
105
- np .array ([255 , 0 , 0 , 255 ], dtype = "uint8" ),
105
+ np .array ([0 , 255 , 0 , 255 ], dtype = "uint8" ),
106
106
np .array ([0 , 0 , 0 , 255 ], dtype = "uint8" ),
107
- ) # Set the points of the frame generated to red.
107
+ ) # Set any non-black pixels to green
108
108
np .putmask (
109
109
diff_im ,
110
- expected_frame_data != np . array ([ 0 , 0 , 0 , 255 ], dtype = "uint8" ) ,
111
- np .array ([0 , 255 , 0 , 255 ], dtype = "uint8" ),
112
- ) # Set the points of the frame generated to green.
110
+ expected_frame_data != frame_data ,
111
+ np .array ([255 , 0 , 0 , 255 ], dtype = "uint8" ),
112
+ ) # Set any different pixels to red
113
113
ax .imshow (diff_im , interpolation = "nearest" )
114
- ax .set_title ("Differences summary : (red = got, green = expected )" )
114
+ ax .set_title ("Differences summary : (green = same, red = different )" )
115
115
116
116
plt .show ()
117
117
plt .savefig (f"{ self .scene } .png" )
0 commit comments