app/test/test_add_word.py and app/test/test_add_word.py: move to the matched word before taking the screenshot.

1 files changed, 3 insertions, 2 deletions

diff --git a/app/test/test_add_word.py b/app/test/test_add_word.py
index 9c87e1c..4613dff 100644
--- a/app/test/test_add_word.py
+++ b/app/test/test_add_word.py
@@ -62,14 +62,15 @@ def test_add_word():
         elem.click()
         
         elems = driver.find_elements_by_xpath("//p[@class='new-word']")
-        elems[0].location_once_scrolled_into_view
-        driver.save_screenshot('./app/test/test_add_word_pic3.png')    
     
         found = 0
         for elem in elems:
             if word in elem.text:
                 found = 1
                 break
+
+        elem.location_once_scrolled_into_view
+        driver.save_screenshot('./app/test/test_add_word_pic3.png')    
     
         assert found == 1
     finally: