setting driver options in set_chrome

selenium/test.py

@@ -9,7 +9,6 @@ class ClearBowlTest(unittest.TestCase):
         super().__init__(*args, **kwargs)
 
         self.driver = self.set_chrome()
-        self.driver.implicitly_wait(15)
         self.base_url = "https://clearbowl.herokuapp.com"
 
     def test_login(self):
@@ -30,7 +29,10 @@ class ClearBowlTest(unittest.TestCase):
         options.add_argument('--no-sandbox')
         options.add_argument('--disable-dev-shm-usage')
 
-        return webdriver.Chrome(options=options)
+        driver = webdriver.Chrome(options=options)
+        driver.implicitly_wait(15)
+
+        return driver
 
     def tearDown(self):
         self.driver.quit()