diff --git a/dm_pix/_src/augment_test.py b/dm_pix/_src/augment_test.py index d41ffb2..7554157 100644 --- a/dm_pix/_src/augment_test.py +++ b/dm_pix/_src/augment_test.py @@ -590,6 +590,8 @@ def test_pad_to_size_chw_values(self): result = augment.pad_to_size( image, target_height=3, target_width=3, channel_axis=0 ) + self.assertNotIsInstance(result, np.bool_) + self.assertNotIsInstance(result, np.number) self.assertEqual(result.shape, (1, 3, 3)) np.testing.assert_array_equal(result[0, 1, 1], 1.0) np.testing.assert_array_equal(result[0, 0, :], 0.0)