def test_read_image_float(self): rval = mnist.read_mnist_images(self.train_images_path, 'float32') assert_equal(rval, self.train_features_mock.astype('float32') / 255.) assert_equal(str(rval.dtype), 'float32')
def test_read_image_bool(self): assert_equal(mnist.read_mnist_images(self.train_images_path, 'bool'), self.train_features_mock >= 128)