def test_as_image(self):
        filename = "testnet.png"
        self.net.as_image(filename=filename)
        file_exists = filename in os.listdir(".") 
        if file_exists:
            os.remove("./" + filename)
        
        assert file_exists, "Can create image file."
dataset/ETHPy150Open abhik/pebl/src/pebl/test/test_network.py/TestNetwork.test_as_image/original