Skip to content

Commit 5553926

Browse files
committed
Update example tests
1 parent b9f4a51 commit 5553926

File tree

4 files changed

+12
-12
lines changed

4 files changed

+12
-12
lines changed

examples/test_agent.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,17 @@ def test_user_agent(self):
88
original_user_agent = user_agent_detected
99
if not self.user_agent:
1010
# Using the built-in user-agent string
11-
print("\n\nUser-Agent:\n%s\n" % user_agent_detected)
11+
self._print("\n\nUser-Agent: %s" % user_agent_detected)
1212
else:
1313
# User-agent was overridden using: --agent=STRING
14-
print("\n\nUser-Agent override:\n%s\n" % user_agent_detected)
15-
print(self.get_text(".ip-address p"))
14+
self._print("\n\nUser-Agent override: %s" % user_agent_detected)
15+
print("\n" + self.get_text(".ip-address p"))
1616
self.sleep(3)
1717

1818
# Now change the user-agent using "execute_cdp_cmd()"
1919
if not self.is_chromium():
2020
msg = "\n* execute_cdp_cmd() is only for Chromium browsers"
21-
print(msg)
21+
self._print(msg)
2222
self.skip(msg)
2323
print("\n--------------------------")
2424
try:
@@ -33,8 +33,8 @@ def test_user_agent(self):
3333
)
3434
self.open("http://whatsmyuseragent.org/")
3535
user_agent_detected = self.get_text(".user-agent p")
36-
print("\nUser-Agent (after override):\n%s\n" % user_agent_detected)
37-
print(self.get_text(".ip-address p"))
36+
self._print("\nUser-Agent override: %s" % user_agent_detected)
37+
print("\n" + self.get_text(".ip-address p") + "\n")
3838
self.sleep(3)
3939
finally:
4040
# Reset the user-agent back to the original

examples/test_chinese_pdf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ def test_chinese_pdf(self):
88

99
# Get and print PDF text
1010
pdf_text = self.get_pdf_text(pdf, page=2)
11-
self._print("\n" + pdf_text)
11+
print("\n" + pdf_text)
1212

1313
# Assert PDF contains the expected text on Page 2
1414
self.assert_pdf_text(pdf, "个测试类", page=2)

examples/test_get_pdf_text.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,4 @@ def test_get_pdf_text(self):
88
"Automate_the_Boring_Stuff_sample_ch17.pdf"
99
)
1010
pdf_text = self.get_pdf_text(pdf, page=1)
11-
self._print("\n" + pdf_text)
11+
print("\n" + pdf_text)

examples/user_agent_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,16 @@ def test_user_agent(self):
88
original_user_agent = user_agent_detected
99
if not self.user_agent:
1010
# Using the built-in user-agent string
11-
print("\n\nUser-Agent = %s\n" % user_agent_detected)
11+
self._print("\n\nUser-Agent = %s\n" % user_agent_detected)
1212
else:
1313
# User-agent was overridden using: --agent=STRING
14-
print("\n\nUser-Agent override = %s\n" % user_agent_detected)
14+
self._print("\n\nUser-Agent override = %s\n" % user_agent_detected)
1515
self.sleep(3)
1616

1717
if not self.is_chromium():
1818
# Skip the rest of the test if not using a Chromium browser
1919
msg = "\n* execute_cdp_cmd() is only for Chromium browsers"
20-
print(msg)
20+
self._print(msg)
2121
self.skip(msg)
2222
try:
2323
# Now change the user-agent using "execute_cdp_cmd()"
@@ -33,7 +33,7 @@ def test_user_agent(self):
3333
)
3434
self.open("https://www.whatsmyua.info/")
3535
user_agent_detected = self.get_text("#custom-ua-string")
36-
print("\nUser-Agent (after override) = %s" % user_agent_detected)
36+
self._print("\nUser-Agent override = %s\n" % user_agent_detected)
3737
self.sleep(3)
3838
finally:
3939
# Reset the user-agent back to the original

0 commit comments

Comments
 (0)