From e58205aa424e2bbd47628e1acc1c00c09153525f Mon Sep 17 00:00:00 2001
From: AnnaChupis <rf2@ukr.net>
Date: Mon, 21 Jun 2021 02:00:38 +0300
Subject: [PATCH] [#$359] $<TEST>:<improve tests structure>

---
 tests/integration/hidden_element_test.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/tests/integration/hidden_element_test.py b/tests/integration/hidden_element_test.py
index a9d1e484..c7f30eb7 100644
--- a/tests/integration/hidden_element_test.py
+++ b/tests/integration/hidden_element_test.py
@@ -31,9 +31,7 @@ def test_should(session_browser):
         '''
     )
 
-    hidden_element = session_browser.element("#hidden")
-
-    hidden_element.should(be.in_dom).should(be.hidden)
+    session_browser.element("#hidden").should(be.in_dom).should(be.hidden)
 
 
 def test_action_on_element_found_from_hidden_element(session_browser):
@@ -48,8 +46,8 @@ def test_action_on_element_found_from_hidden_element(session_browser):
         <h2 id="second">Heading 2</h2>
         '''
     )
-    hidden_element = session_browser.element("#hidden")
 
-    hidden_element.element('./following-sibling::*/a[2]').click()
+    session_browser.element("#hidden").\
+        element('./following-sibling::*/a[2]').click()
 
     assert 'second' in session_browser.driver.current_url