MNIST tests work-in-progress