From ec2a95adf682ddfbc7cd374be4150ced32628b83 Mon Sep 17 00:00:00 2001
From: torzdf <36920800+torzdf@users.noreply.github.com>
Date: Wed, 17 Apr 2024 14:02:43 +0100
Subject: [PATCH] tests: Fix alignment tool process_video test

---
 tests/tools/alignments/media_test.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/tools/alignments/media_test.py b/tests/tools/alignments/media_test.py
index 5f603ee3..17e45a51 100644
--- a/tests/tools/alignments/media_test.py
+++ b/tests/tools/alignments/media_test.py
@@ -625,12 +625,13 @@ class TestFrames:
         folder : str
             Dummy media folder
         """
-        expected = [{"frame_fullname": "images_000001.png",
+        ext = os.path.splitext(folder)[-1]
+        expected = [{"frame_fullname": f"images_000001{ext}",
                      "frame_name": "images_000001",
-                     "frame_extension": ".png"},
-                    {"frame_fullname": "images_000002.png",
+                     "frame_extension": ext},
+                    {"frame_fullname": f"images_000002{ext}",
                      "frame_name": "images_000002",
-                     "frame_extension": ".png"}]
+                     "frame_extension": ext}]
 
         frames = Frames(folder, None)
         returned = list(frames.process_video())