increment

Author: tuttlepower

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="predictit_markets",  # Package name
-    version="0.1.4", 
+    version="0.1.5", 
     author="Travis Tuttle",
     author_email="28586132+tuttlepower@users.noreply.github.com",
     description="A package for interacting with PredictIt markets",

tests/test_predictit_markets.py

@@ -6,15 +6,28 @@
 import pandas as pd
 from predictit_markets.predictit_markets import all_markets, market_data
 
+
 class TestPredictItMarkets(unittest.TestCase):
 
-    @patch('predictit_markets.predictit_markets.requests.get')
+    @patch("predictit_markets.predictit_markets.requests.get")
     def test_market_data_success(self, mock_get):
         # Simulate a successful API response
         mock_response = Mock()
         mock_response.status_code = 200
         mock_response.json.return_value = [
-            {'marketId': 8089, 'contractId': 31414, 'contractName': 'Shapiro', 'date': '2024-07-26T00:00:00', 'dateString': '2024-07-26', 'openSharePrice': 0.26, 'highSharePrice': 0.34, 'lowSharePrice': 0.25, 'closeSharePrice': 0.31, 'tradeVolume': 160257, 'lineColor': '#0D8ECF'}
+            {
+                "marketId": 8089,
+                "contractId": 31414,
+                "contractName": "Shapiro",
+                "date": "2024-07-26T00:00:00",
+                "dateString": "2024-07-26",
+                "openSharePrice": 0.26,
+                "highSharePrice": 0.34,
+                "lowSharePrice": 0.25,
+                "closeSharePrice": 0.31,
+                "tradeVolume": 160257,
+                "lineColor": "#0D8ECF",
+            }
             # Add more sample data here if needed
         ]
         mock_get.return_value = mock_response
@@ -25,13 +38,15 @@ def test_market_data_success(self, mock_get):
         # Assert the DataFrame is not None and contains the expected data
         self.assertIsNotNone(df)
         self.assertEqual(len(df), 1)
-        self.assertEqual(df.iloc[0]['contractName'], 'Shapiro')
+        self.assertEqual(df.iloc[0]["contractName"], "Shapiro")
 
-    @patch('predictit_markets.predictit_markets.requests.get')
+    @patch("predictit_markets.predictit_markets.requests.get")
     def test_market_data_http_error(self, mock_get):
         # Simulate an HTTP error response
         mock_response = Mock()
-        mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError("404 Client Error: Not Found")
+        mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError(
+            "404 Client Error: Not Found"
+        )
         mock_get.return_value = mock_response
 
         # Test the market_data function
@@ -40,7 +55,7 @@ def test_market_data_http_error(self, mock_get):
         # Assert the DataFrame is None due to the error
         self.assertIsNone(df)
 
-    @patch('predictit_markets.predictit_markets.requests.get')
+    @patch("predictit_markets.predictit_markets.requests.get")
     def test_market_data_request_exception(self, mock_get):
         # Simulate a request exception
         mock_get.side_effect = requests.exceptions.RequestException("Network error")
@@ -51,7 +66,7 @@ def test_market_data_request_exception(self, mock_get):
         # Assert the DataFrame is None due to the exception
         self.assertIsNone(df)
 
-    @patch('predictit_markets.predictit_markets.requests.get')
+    @patch("predictit_markets.predictit_markets.requests.get")
     def test_market_data_json_error(self, mock_get):
         # Simulate a JSON decoding error
         mock_response = Mock()
@@ -64,27 +79,28 @@ def test_market_data_json_error(self, mock_get):
 
         # Assert the DataFrame is None due to JSON error
         self.assertIsNone(df)
-    
-    @patch('predictit_markets.requests.get')
+
+    @patch("predictit_markets.requests.get")
     def test_all_markets_success(self, mock_get):
         mock_get.return_value.status_code = 200
         mock_get.return_value.json.return_value = {
-        'markets': [
-            {'id': 123, 'name': 'Market 1'},
-            {'id': 456, 'name': 'Market 2'},
+            "markets": [
+                {"id": 123, "name": "Market 1"},
+                {"id": 456, "name": "Market 2"},
             ]
         }
 
         df = all_markets()
 
-   
-        expected_df = pd.DataFrame([
-        {'Market ID': 123, 'Market Name': 'Market 1'},
-        {'Market ID': 456, 'Market Name': 'Market 2'},
-    ])
+        expected_df = pd.DataFrame(
+            [
+                {"Market ID": 123, "Market Name": "Market 1"},
+                {"Market ID": 456, "Market Name": "Market 2"},
+            ]
+        )
 
         pd.testing.assert_frame_equal(df, expected_df)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()