use setupclass (#130)

Author: clickingbuttons

tests/mocks.py

@@ -23,13 +23,9 @@
 
 
 class BaseTest(unittest.TestCase):
-    setup = False
-
-    def setUp(self):
-        if self.setup:
-            return
+    @classmethod
+    def setUpClass(cls):
+        cls.c = RESTClient("")
         httpretty.enable(verbose=True, allow_net_connect=False)
-        c = RESTClient("")
         for m in mocks:
-            httpretty.register_uri(httpretty.GET, c.BASE + m[0], m[1])
-        self.setup = True
+            httpretty.register_uri(httpretty.GET, cls.c.BASE + m[0], m[1])

tests/test_aggs.py

@@ -1,17 +1,15 @@
-from polygon import RESTClient
+from mocks import BaseTest
 from polygon.rest.models import (
     Agg,
     GroupedDailyAgg,
     DailyOpenCloseAgg,
     PreviousCloseAgg,
 )
-from mocks import BaseTest
 
 
 class AggsTest(BaseTest):
     def test_get_aggs(self):
-        c = RESTClient("")
-        aggs = c.get_aggs("AAPL", 1, "day", "2005-04-01", "2005-04-04")
+        aggs = self.c.get_aggs("AAPL", 1, "day", "2005-04-01", "2005-04-04")
         expected = [
             Agg(
                 open=1.5032,
@@ -37,8 +35,7 @@ def test_get_aggs(self):
         self.assertEqual(aggs, expected)
 
     def test_get_grouped_daily_aggs(self):
-        c = RESTClient("")
-        aggs = c.get_grouped_daily_aggs("2005-04-04", True)
+        aggs = self.c.get_grouped_daily_aggs("2005-04-04", True)
         expected = [
             GroupedDailyAgg(
                 ticker="GIK",
@@ -55,8 +52,7 @@ def test_get_grouped_daily_aggs(self):
         self.assertEqual(aggs, expected)
 
     def test_get_daily_open_close_agg(self):
-        c = RESTClient("")
-        aggs = c.get_daily_open_close_agg("AAPL", "2005-04-01", True)
+        aggs = self.c.get_daily_open_close_agg("AAPL", "2005-04-01", True)
         expected = [
             DailyOpenCloseAgg(
                 after_hours=123,
@@ -74,8 +70,7 @@ def test_get_daily_open_close_agg(self):
         self.assertEqual(aggs, expected)
 
     def test_get_previous_close_agg(self):
-        c = RESTClient("")
-        aggs = c.get_previous_close_agg("AAPL")
+        aggs = self.c.get_previous_close_agg("AAPL")
         expected = [
             PreviousCloseAgg(
                 ticker="AAPL",