mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-02 03:31:52 +08:00
Victor Grau Serrat
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
from catalyst.exchange.poloniex.poloniex import Poloniex
|
|
from catalyst.finance.order import Order
|
|
from base import BaseExchangeTestCase
|
|
from logbook import Logger
|
|
from catalyst.exchange.exchange_utils import get_exchange_auth
|
|
import pandas as pd
|
|
from test_utils import output_df
|
|
|
|
log = Logger('test_poloniex')
|
|
|
|
|
|
class TestPoloniex(BaseExchangeTestCase):
|
|
@classmethod
|
|
def setup(self):
|
|
print ('creating poloniex object')
|
|
auth = get_exchange_auth('poloniex')
|
|
self.exchange = Poloniex(
|
|
key=auth['key'],
|
|
secret=auth['secret'],
|
|
base_currency='btc'
|
|
)
|
|
|
|
def test_order(self):
|
|
log.info('creating order')
|
|
asset = self.exchange.get_asset('neos_btc')
|
|
order_id = self.exchange.order(
|
|
asset=asset,
|
|
limit_price=0.0005,
|
|
amount=1,
|
|
)
|
|
log.info('order created {}'.format(order_id))
|
|
assert order_id is not None
|
|
pass
|
|
|
|
def test_open_orders(self):
|
|
log.info('retrieving open orders')
|
|
# asset = self.exchange.get_asset('neos_btc')
|
|
# orders = self.exchange.get_open_orders(asset)
|
|
pass
|
|
|
|
def test_get_order(self):
|
|
log.info('retrieving order')
|
|
order = self.exchange.get_order(
|
|
u'2c584020-9caf-4af5-bde0-332c0bba17e2')
|
|
assert isinstance(order, Order)
|
|
pass
|
|
|
|
def test_cancel_order(self, ):
|
|
log.info('cancel order')
|
|
self.exchange.cancel_order(u'dc7bcca2-5219-4145-8848-8a593d2a72f9')
|
|
pass
|
|
|
|
def test_get_candles(self):
|
|
log.info('retrieving candles')
|
|
assets = self.exchange.get_asset('eth_btc')
|
|
ohlcv = self.exchange.get_candles(
|
|
# end_dt=pd.to_datetime('2017-11-01', utc=True),
|
|
end_dt=None,
|
|
freq='5T',
|
|
assets=assets,
|
|
bar_count=200
|
|
)
|
|
df = pd.DataFrame(ohlcv)
|
|
df.set_index('last_traded', drop=True, inplace=True)
|
|
log.info(df.tail(25))
|
|
|
|
path = output_df(df, assets, '5min_candles')
|
|
log.info('saved candles: {}'.format(path))
|
|
pass
|
|
|
|
def test_tickers(self):
|
|
log.info('retrieving tickers')
|
|
tickers = self.exchange.tickers([
|
|
self.exchange.get_asset('eth_btc'),
|
|
self.exchange.get_asset('etc_btc')
|
|
])
|
|
assert len(tickers) == 2
|
|
pass
|
|
|
|
def test_get_balances(self):
|
|
log.info('testing wallet balances')
|
|
# balances = self.exchange.get_balances()
|
|
pass
|
|
|
|
def test_get_account(self):
|
|
log.info('testing account data')
|
|
pass
|
|
|
|
def test_orderbook(self):
|
|
log.info('testing order book for poloniex')
|
|
# asset = self.exchange.get_asset('eth_btc')
|
|
# orderbook = self.exchange.get_orderbook(asset)
|
|
pass
|