diff --git a/seleniumbase/console_scripts/sb_mkdir.py b/seleniumbase/console_scripts/sb_mkdir.py index 958f0fe3..abacb139 100755 --- a/seleniumbase/console_scripts/sb_mkdir.py +++ b/seleniumbase/console_scripts/sb_mkdir.py @@ -320,7 +320,6 @@ def main(): data.append("") data.append("") data.append("class DemoSiteTests(BaseCase):") - data.append("") data.append(" def test_demo_site(self):") data.append(' self.open("https://seleniumbase.io/demo_page.html")') data.append(' self.assert_title("Web Testing Page")') @@ -391,19 +390,20 @@ def main(): data.append("") data.append("") data.append("class GoogleTests(BaseCase):") - data.append("") - data.append(" @parameterized.expand([") - data.append(' ["pypi", "pypi.org"],') - data.append(' ["wikipedia", "wikipedia.org"],') - data.append(' ["seleniumbase", "seleniumbase/SeleniumBase"],') - data.append(" ])") + data.append(" @parameterized.expand(") + data.append(" [") + data.append(' ["pypi", "pypi.org"],') + data.append(' ["wikipedia", "wikipedia.org"],') + data.append(' ["seleniumbase", "seleniumbase/SeleniumBase"],') + data.append(" ]") + data.append(" )") data.append(" def test_parameterized_google_search(" "self, search_term, expected_text):") - data.append(" self.open('https://google.com/ncr')") + data.append(' self.open("https://google.com/ncr")') data.append(" self.type('input[title=\"Search\"]', " - "search_term + '\\n')") - data.append(" self.assert_element('#result-stats')") - data.append(" self.assert_text(expected_text, '#search')") + 'search_term + "\\n")') + data.append(' self.assert_element("#result-stats")') + data.append(' self.assert_text(expected_text, "#search")') data.append("") file_path = "%s/%s" % (dir_name, "parameterized_test.py") file = codecs.open(file_path, "w+", "utf-8") @@ -425,7 +425,6 @@ def main(): data.append("") data.append("") data.append("class BaseTestCase(BaseCase):") - data.append("") data.append(" def setUp(self):") if sys.version_info[0] >= 3: data.append(" super().setUp()") @@ -476,7 +475,6 @@ def main(): data.append("") data.append("") data.append("class MyTestClass(BaseTestCase):") - data.append("") data.append(" def test_boilerplate(self):") data.append(" self.login()") data.append(" self.example_method()") @@ -491,8 +489,7 @@ def main(): data.append("from seleniumbase import BaseCase") data.append("") data.append("") - data.append("class DataPage():") - data.append("") + data.append("class DataPage:") data.append(" def go_to_data_url(self, sb):") data.append(' sb.open("data:text/html,

Hello!

")') data.append("") @@ -501,7 +498,6 @@ def main(): data.append("") data.append("") data.append("class ObjTests(BaseCase):") - data.append("") data.append(" def test_data_url_page(self):") data.append(" DataPage().go_to_data_url(self)") data.append(' self.assert_text("Hello!", "p")') @@ -513,8 +509,7 @@ def main(): file.close() data = [] - data.append("class DataPage():") - data.append("") + data.append("class DataPage:") data.append(" def go_to_data_url(self, sb):") data.append(' sb.open("data:text/html,

Hello!

")') data.append("") @@ -522,8 +517,7 @@ def main(): data.append(' sb.type("input", text)') data.append("") data.append("") - data.append("class ObjTests():") - data.append("") + data.append("class ObjTests:") data.append(" def test_data_url_page(self, sb):") data.append(" DataPage().go_to_data_url(sb)") data.append(' sb.assert_text("Hello!", "p")') @@ -550,17 +544,16 @@ def main(): data.append("") data.append("") data.append("class GoogleTests(BaseCase):") - data.append("") data.append(" def test_google_dot_com(self):") - data.append(" self.open('https://google.com/ncr')") - data.append(" self.type(HomePage.search_box, 'github')") + data.append(' self.open("https://google.com/ncr")') + data.append(' self.type(HomePage.search_box, "github")') data.append(" self.assert_element(HomePage.list_box)") data.append(" self.assert_element(HomePage.search_button)") data.append( " self.assert_element(HomePage.feeling_lucky_button)") data.append(" self.click(HomePage.search_button)") data.append( - " self.assert_text('github.com', " + ' self.assert_text("github.com", ' "ResultsPage.search_results)") data.append(" self.assert_element(ResultsPage.images_link)") data.append("") @@ -577,13 +570,13 @@ def main(): data.append(" search_button = 'input[value=\"Google Search\"]'") data.append( " feeling_lucky_button = " - "'''input[value=\"I'm Feeling Lucky\"]'''") + '"""input[value=\"I\'m Feeling Lucky\"]"""') data.append("") data.append("") data.append("class ResultsPage(object):") data.append(" google_logo = 'img[alt=\"Google\"]'") - data.append(" images_link = 'link=Images'") - data.append(" search_results = 'div#center_col'") + data.append(' images_link = "link=Images"') + data.append(' search_results = "div#center_col"') data.append("") file_path = "%s/%s" % (dir_name_3, "google_objects.py") file = codecs.open(file_path, "w+", "utf-8") @@ -594,8 +587,7 @@ def main(): data.append("from seleniumbase import BaseCase") data.append("") data.append("") - data.append("class LoginPage():") - data.append("") + data.append("class LoginPage:") data.append(" def login_to_swag_labs(self, sb, username):") data.append(' sb.open("https://www.saucedemo.com/")') data.append(' sb.type("#user-name", username)') @@ -604,7 +596,6 @@ def main(): data.append("") data.append("") data.append("class MyTests(BaseCase):") - data.append("") data.append(" def test_swag_labs_login(self):") data.append( ' LoginPage().login_to_swag_labs(self, "standard_user")') @@ -617,8 +608,7 @@ def main(): file.close() data = [] - data.append("class LoginPage():") - data.append("") + data.append("class LoginPage:") data.append(" def login_to_swag_labs(self, sb, username):") data.append(' sb.open("https://www.saucedemo.com/")') data.append(' sb.type("#user-name", username)') @@ -626,8 +616,7 @@ def main(): data.append(" sb.click('input[type=\"submit\"]')") data.append("") data.append("") - data.append("class MyTests():") - data.append("") + data.append("class MyTests:") data.append(" def test_swag_labs_login(self, sb):") data.append(' LoginPage().login_to_swag_labs(sb, "standard_user")') data.append(' sb.assert_element("#inventory_container")')