From 03a1de2b6564fb17e67648912d02413539ac2828 Mon Sep 17 00:00:00 2001
From: Ryan Frederich <ryanfrederich@gmail.com>
Date: Wed, 21 Aug 2024 13:21:29 -0700
Subject: [PATCH] lint fixes

---
 tests/test_art.py    | 14 ++++++++------
 tests/test_gpt.py    |  1 +
 tests/test_helper.py |  8 +++-----
 3 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/tests/test_art.py b/tests/test_art.py
index e99450a..4d9c4f9 100644
--- a/tests/test_art.py
+++ b/tests/test_art.py
@@ -4,27 +4,29 @@
 Run pytest: pytest
 """
 
-import sys
 import io
+import sys
+
 from src import art
 
+
 def test_print_wave():
     """
     Testing the print_wave() function
     Uses sys & io to capture printed output
     """
     # Capture the output
-    captured_output = io.StringIO()          # Create StringIO object
-    sys.stdout = captured_output             # Redirect stdout.
+    captured_output = io.StringIO()  # Create StringIO object
+    sys.stdout = captured_output  # Redirect stdout.
 
     # Call the function
     art.print_wave(1, 0, "blue")
 
     # Reset redirect.
-    sys.stdout = sys.__stdout__     
+    sys.stdout = sys.__stdout__
 
     # Now captured_output.getvalue() contains the printed content
     output = captured_output.getvalue()
-    
+
     # Perform assertions based on expected output
-    assert output != "", "print_wave() did not print anything"
\ No newline at end of file
+    assert output
diff --git a/tests/test_gpt.py b/tests/test_gpt.py
index 3292319..c683423 100644
--- a/tests/test_gpt.py
+++ b/tests/test_gpt.py
@@ -6,6 +6,7 @@
 
 from src import gpt
 
+
 def test_simple_gpt():
     """
     Testing the simple_gpt function
diff --git a/tests/test_helper.py b/tests/test_helper.py
index 6287cf2..15319ea 100644
--- a/tests/test_helper.py
+++ b/tests/test_helper.py
@@ -7,8 +7,7 @@
 import io
 from unittest.mock import patch
 
-from src import cli
-from src import helper
+from src import cli, helper
 
 
 def test_invalid_input():
@@ -44,6 +43,7 @@ def test_json_output():
     assert type(json_output["Lat"]) in {int, float}
     assert isinstance(json_output["Location"], str)
 
+
 def test_print_gpt():
     """
     Tests the simple_gpt()
@@ -53,11 +53,9 @@ def test_print_gpt():
         "Height": "test",
         "Swell Direction": "test",
         "Period": "test",
-        "Unit": "test"
+        "Unit": "test",
     }
     gpt_prompt = "Please output 'gpt works'"
     gpt_info = [None, ""]
     gpt_response = helper.print_gpt(surf_data, gpt_prompt, gpt_info)
     assert "gpt works" in gpt_response
-
-