Update test_h5image.py

test_h5image.py

@@ -13,7 +13,7 @@ map_array = np.array(h5_image.get_map(map_name))
 print("map_array.shape", map_array.shape, "type", type(map_array) )
 
 legend = h5_image.get_layers(map_name)[2]
-legend_patch = h5_image.get_legend(map_name, legend)
+legend_patch = h5_image.get_legend(map_name, 'IPca_poly')
 legend_resized = tf.image.resize(legend_patch, (256,256))
 
 # Convert to uint8 range [0, 255] if necessary