diff --git a/tdameritrade/orders/constants.py b/tdameritrade/orders/constants.py index 3209212..7cf5c69 100644 --- a/tdameritrade/orders/constants.py +++ b/tdameritrade/orders/constants.py @@ -1,5 +1,6 @@ """Constants used in Orders """ + from enum import Enum diff --git a/tdameritrade/orders/models/leg.py b/tdameritrade/orders/models/leg.py index 5b43cc0..18ee410 100644 --- a/tdameritrade/orders/models/leg.py +++ b/tdameritrade/orders/models/leg.py @@ -1,6 +1,7 @@ """ Order Leg usedin in Orders API """ + from dataclasses import dataclass from ..constants import ( diff --git a/tdameritrade/tests/test_client.py b/tdameritrade/tests/test_client.py index c1bf69f..cc6d83e 100644 --- a/tdameritrade/tests/test_client.py +++ b/tdameritrade/tests/test_client.py @@ -15,24 +15,6 @@ def tdclient(): class TestExtension: - def setup(self): - pass - # setup() before each test method - - def teardown(self): - pass - # teardown() after each test method - - @classmethod - def setup_class(cls): - pass - # setup_class() before any methods in this class - - @classmethod - def teardown_class(cls): - pass - # teardown_class() after any methods in this class - def test_init(self, tdclient): assert tdclient._refreshToken == "reftoken"