def test_returnOfCapital(self):
from pyEX import returnOfCapital
with patch('requests.get') as mock, \
patch('pickle.dump'):
mock.return_value = MagicMock()
mock.return_value.status_code = 200
returnOfCapital()
returnOfCapital(SYMBOL)
returnOfCapital(SYMBOL, 'test')
def test_returnOfCapital(self):
from pyEX import returnOfCapital
with patch("requests.get") as mock, patch("pickle.dump"):
mock.return_value = MagicMock()
mock.return_value.status_code = 200
returnOfCapital()
returnOfCapital(SYMBOL)
returnOfCapital(SYMBOL, "test")
