ENH: Changing test to use images on github repo

ARM-DOE · Dec 19, 2023 · c5d1763 · c5d1763
1 parent ec0642a
commit c5d1763
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/act/tests/utils/test_io_utils.py b/act/tests/utils/test_io_utils.py
@@ -5,6 +5,7 @@
 import random
 import glob
 import pytest
+import os
 
 import act
 from act.tests import sample_files
@@ -215,8 +216,11 @@ def test_gunzip():
 
 @pytest.mark.skipif(not MOVIEPY_AVAILABLE, reason='MoviePy is not installed.')
 def test_generate_movie():
-    act_dir = act.__file__
-    baseline = '/'.join(act_dir.split('/')[0:-1] + ['tests', 'plotting', 'baseline'])
-    files = glob.glob(baseline + '/*contour*png')
-    result = act.utils.generate_movie(files)
+    files = [
+        'https://github.com/ARM-DOE/ACT/blob/main/act/tests/plotting/baseline/test_contour.png?raw=true',
+        'https://github.com/ARM-DOE/ACT/blob/main/act/tests/plotting/baseline/test_contour2.png?raw=true',
+        'https://github.com/ARM-DOE/ACT/blob/main/act/tests/plotting/baseline/test_contourf.png?raw=true',
+        'https://github.com/ARM-DOE/ACT/blob/main/act/tests/plotting/baseline/test_contourf2.png?raw=true'
+    ]
+    result = act.utils.generate_movie(files, fps=5)
     assert str(result) == 'movie.mp4'