6273d5bcf7292ca77d7730bcd7102655f7c3bb82,tensorforce/tests/test_preprocessing.py,TestPreprocessing,test_preprocessing_grayscale,#TestPreprocessing#,47

Before Change

        
        pp = tensorforce.core.preprocessing.grayscale.Grayscale()

        shape = list(np.random.randint(1, 8, size=2)) + [3]

        processed_state, processed_shape = self._test_preprocessing_shape(pp, shape, tuple(shape[0:2]))

After Change

    def test_preprocessing_grayscale(self):
        grayscale = Grayscale()
        shape = (randint(1, 64), randint(1, 64), 3)
        state = np.random.randint(0, 256, shape)
        self.assertEqual(grayscale.processed_shape(shape), (shape[0], shape[1], 1))
        self.assertEqual(grayscale.process(state).shape, grayscale.processed_shape(shape))

Instances