761e760651f7381c8bde06b2b3a00a93decde180,art/visualization_unittest.py,TestVisualization,test_save_image,#TestVisualization#,16

Before Change

class TestVisualization(unittest.TestCase):

    def test_save_image(self):
        (x, _), (_, _), _, _ = load_mnist_raw()

        f_name = "image1.png"
        save_image(x[0], f_name)
        path = os.path.join(DATA_PATH, f_name)

After Change

class TestVisualization(unittest.TestCase):

    def test_save_image(self):
        (x, _), (_, _), _, _ = load_mnist(raw=True)

        f_name = "image1.png"
        save_image(x[0], f_name)
        path = os.path.join(DATA_PATH, f_name)

Instances