File size: 459 Bytes
ee3a6ac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import unittest
from lib.gas import save_image_to_drive

class TestGAS(unittest.TestCase):
    def test_save_image_to_drive(self):
        # Load a sample image
        with open("sample_image.jpg", "rb") as f:
            image_data = f.read()

        # Save the image to Google Drive
        save_image_to_drive(image_data)

        # Assert that the file was saved successfully
        self.assertTrue(True)

if __name__ == "__main__":
    unittest.main()