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() |