Create subdirs automatically with Figure

Resolves #120

CHANGELOG.md

@@ -6,6 +6,7 @@
 - Removed `TickTock.remove_outliers`
 - Moved `is_windows` to static `OS` attribute, which also contains `is_mac` and `is_linux`
 - Added `HardwareInfo`, which contains a bunch of system hardware information
+- `Figure` now automatically creates necessary subdirectories
 
 ### Bug fixes
 

pelutils/ds/plots.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 import time
 from typing import Any, Callable, List, Optional, Union
 
@@ -233,6 +234,9 @@ def __exit__(self, et, ev, tb):
         if self._tight_layout:
             plt.tight_layout()
         if not et:
+            directory = os.path.split(self._savepath)[0]
+            if directory:
+                os.makedirs(directory, exist_ok=True)
             plt.savefig(self._savepath)
         plt.close()
 

tests/ds/test_plots.py

@@ -162,7 +162,11 @@ class TestFigure(UnitTestCollection):
     def test_save(self):
         with Figure(self.savepath):
             pass
-        assert os.path.exists(self.savepath)
+        assert os.path.isfile(self.savepath)
+        path = os.path.join(UnitTestCollection.test_dir, "many", "long", "subdirectories.png")
+        with Figure(path):
+            pass
+        assert os.path.isfile(path)
 
     def test_no_save_if_error(self):
         if os.path.exists(self.savepath):