Merge pull request NixOS#222557 from alyssais/nixos-test-driver-scree…

…nshot-charset nixos/test-driver: accept non-\w* filenames
lopsided98 · Mar 23, 2023 · 09ad6a7 · 09ad6a7
2 parents 0a528a3 + 27f9795
commit 09ad6a7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nixos/lib/test-driver/test_driver/machine.py b/nixos/lib/test-driver/test_driver/machine.py
@@ -737,9 +737,10 @@ def connect(self) -> None:
             self.connected = True
 
     def screenshot(self, filename: str) -> None:
-        word_pattern = re.compile(r"^\w+$")
-        if word_pattern.match(filename):
-            filename = os.path.join(self.out_dir, f"{filename}.png")
+        if "." not in filename:
+            filename += ".png"
+        if "/" not in filename:
+            filename = os.path.join(self.out_dir, filename)
         tmp = f"{filename}.ppm"
 
         with self.nested(