def test_baboon(self):
        assert_equals(baboon().dtype, 'float32')
        assert_equals(baboon().shape, (1, 512, 512))
        assert_equals(baboon(size=(128, 256)).shape, (1, 128, 256))
        assert_equals(baboon(size=64).shape, (1, 64, 64))