From a550b7d27babbfd5ea0831e687f9bf987db57fe0 Mon Sep 17 00:00:00 2001
From: Roland <roland@learningselfreliance.com>
Date: Thu, 3 Feb 2022 13:16:10 -0700
Subject: [PATCH] Using lib in otp tests.

---
 modules/otp/test/test_otp.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/modules/otp/test/test_otp.py b/modules/otp/test/test_otp.py
index f97ea89f..f533c95e 100644
--- a/modules/otp/test/test_otp.py
+++ b/modules/otp/test/test_otp.py
@@ -1,6 +1,6 @@
 import pytest
 
-from modules.otp.lib import format_message, encrypt_otp, decrypt_otp, generate_pdf
+from modules.otp import lib
 
 
 @pytest.mark.parametrize(
@@ -25,7 +25,7 @@ from modules.otp.lib import format_message, encrypt_otp, decrypt_otp, generate_p
     ],
 )
 def test_encrypt_otp(otp, plaintext, ciphertext):
-    result = encrypt_otp(otp, plaintext)
+    result = lib.encrypt_otp(otp, plaintext)
     expected = dict(
         otp=otp,
         plaintext=plaintext,
@@ -36,7 +36,7 @@ def test_encrypt_otp(otp, plaintext, ciphertext):
     # Whitespace is ignored
     plaintext = f'\t   {plaintext}'
     otp = f'\n {otp} '
-    result = encrypt_otp(otp, plaintext)
+    result = lib.encrypt_otp(otp, plaintext)
     expected = dict(
         otp=otp.strip(),
         plaintext=plaintext.strip(),
@@ -63,7 +63,7 @@ def test_encrypt_otp(otp, plaintext, ciphertext):
     ],
 )
 def test_decrypt_otp(otp, ciphertext, plaintext):
-    result = decrypt_otp(otp, ciphertext)
+    result = lib.decrypt_otp(otp, ciphertext)
     expected = dict(
         otp=otp,
         plaintext=plaintext,
@@ -92,7 +92,7 @@ def test_decrypt_otp(otp, ciphertext, plaintext):
     ]
 )
 def test_format_message(message, expected):
-    result = format_message(message)
+    result = lib.format_message(message)
     assert result == expected
 
 
@@ -100,4 +100,4 @@ def test_latex_pdf():
     """
     Generate a OTP PDF.
     """
-    assert generate_pdf()
+    assert lib.generate_pdf()