From d8b29ff6d26b47b0fade368eeca6ff8e7d9d8603 Mon Sep 17 00:00:00 2001
From: Akash <64683866+akamhy@users.noreply.github.com>
Date: Fri, 24 Jul 2020 14:54:27 +0530
Subject: [PATCH] Update test_wrapper.py

---
 tests/test_wrapper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py
index b4f3ba5..0be8888 100644
--- a/tests/test_wrapper.py
+++ b/tests/test_wrapper.py
@@ -26,7 +26,7 @@ def test_clean_url():
 def test_dunders():
     url = "https://en.wikipedia.org/wiki/Network_security"
     user_agent = "UA"
-    target = waybackpy.Url(test_url, user_agent)
+    target = waybackpy.Url(url, user_agent)
     assert "waybackpy.Url(url=%s, user_agent=%s)" % (url, user_agent) == repr(target)
     assert len(target) == len(url)
     assert str(target) == url
@@ -35,7 +35,7 @@ def test_archive_url_parser():
     request_url = "https://amazon.com"
     hdr = {"User-Agent": user_agent}  # nosec
     req = Request(request_url, headers=hdr)  # nosec
-    header = _get_response(req).headers
+    header = waybackpy._get_response(req).headers
     with pytest.raises(Exception):
         waybackpy._archive_url_parser(header)