diff --git a/tests/mocks.py b/tests/mocks.py index a00bae89..02e2b15c 100644 --- a/tests/mocks.py +++ b/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]) diff --git a/tests/test_aggs.py b/tests/test_aggs.py index 961df02b..5e651786 100644 --- a/tests/test_aggs.py +++ b/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",