Pass in base and source separately when setting save directory.

Structure is important for comparison script.
bbangert · Mar 13, 2014 · 5101982 · 5101982
1 parent 94f24f9
commit 5101982
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lettuce_webdriver/screenshot.py b/lettuce_webdriver/screenshot.py
@@ -6,11 +6,12 @@
 )
 import os.path
 
-def set_save_directory(root):
+def set_save_directory(base, source):
     """Sets the root save directory for saving screenshots.
     
     Screenshots will be saved in subdirectories under this directory by
     browser window size. """
+    root = os.path.join(base, source)
     if not os.path.isdir(root):
         os.makedirs(root)