File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change 77
88
99class Tests (IntegrationTests ):
10- def create_usage_test (self , filename ):
10+ def create_usage_test (self , filename , dir_name = 'usage' ):
1111 app = importlib .import_module (filename ).app
1212
1313 self .startServer (app )
@@ -16,9 +16,20 @@ def create_usage_test(self, filename):
1616 EC .presence_of_element_located ((By .ID , "cytoscape" ))
1717 )
1818
19+ directory_path = os .path .join (
20+ os .path .dirname (__file__ ),
21+ 'screenshots' ,
22+ dir_name
23+ )
24+
25+ # Create directory if it doesn't already exist
26+ if not os .path .exists (directory_path ):
27+ os .makedirs (directory_path )
28+
1929 self .driver .save_screenshot (os .path .join (
2030 os .path .dirname (__file__ ),
2131 'screenshots' ,
32+ dir_name ,
2233 filename + '.png'
2334 ))
2435
You can’t perform that action at this time.
0 commit comments