def test_correct_output_shape(self):
    img = (np.random.rand(928, 928, 3) * 255)
    v = Visualizer(img, self.metadata)
    out = v.output.get_image()
    self.assertEqual(out.shape, img.shape)
def test_correct_output_shape(self):
    img = (np.random.rand(928, 928, 3) * 255)
    v = Visualizer(img, self.metadata)
    out = v.output.get_image()
    self.assertEqual(out.shape, img.shape)