coinbasepro-python/tests/test_public_client.py at 3ad742e507cd90306ac8f13359fb0d2b8ae0a767 · danpaquin/coinbasepro-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import pytest
import time
from itertools import islice
import datetime
from dateutil.relativedelta import relativedelta
from cbpro.public_client import PublicClient


@pytest.fixture(scope='module')
def client():
    return PublicClient()


@pytest.mark.usefixtures('client')
class TestPublicClient(object):

    @staticmethod
    def teardown_method():
        time.sleep(.25)  # Avoid rate limit

    def test_get_products(self, client):
        r = client.get_products()
        assert type(r) is list

    def test_get_single_product(self, client):
        r = client.get_single_product("BTC-USD")
        assert isinstance(r, dict)

    @pytest.mark.parametrize('level', [1, 2, 3, None])
    def test_get_product_order_book(self, client, level):
        r = client.get_product_order_book('BTC-USD', level=level)
        assert type(r) is dict
        assert 'asks' in r
        assert 'bids' in r

        if level in (1, None) and (len(r['asks']) > 1 or len(r['bids']) > 1):
            pytest.fail('Fail: Level 1 should only return the best ask and bid')

        if level is 2 and (len(r['asks']) > 50 or len(r['bids']) > 50):
            pytest.fail('Fail: Level 2 should only return the top 50 asks and bids')

        if level is 3 and (len(r['asks']) < 50 or len(r['bids']) < 50):
            pytest.fail('Fail: Level 3 should return the full order book')

    def test_get_product_ticker(self, client):
        r = client.get_product_ticker('BTC-USD')
        assert type(r) is dict
        assert 'ask' in r
        assert 'trade_id' in r

    def test_get_product_trades(self, client):
        r = list(islice(client.get_product_trades('BTC-USD'), 200))
        assert type(r) is list
        assert 'trade_id' in r[0]

    current_time = datetime.datetime.now()

    @pytest.mark.parametrize('start,end,granularity',
                             [(current_time - relativedelta(months=1),
                               current_time, 21600)])
    def test_get_historic_rates(self, client, start, end, granularity):
        r = client.get_product_historic_rates('BTC-USD', start=start, end=end, granularity=granularity)
        assert type(r) is list
        for ticker in r:
            assert( all( [type(x) in (int, float) for x in ticker ] ) )

    def test_get_product_24hr_stats(self, client):
        r = client.get_product_24hr_stats('BTC-USD')
        assert type(r) is dict
        assert 'volume_30day' in r

    def test_get_currencies(self, client):
        r = client.get_currencies()
        assert type(r) is list
        assert 'name' in r[0]

    def test_get_time(self, client):
        r = client.get_time()
        assert type(r) is dict
        assert 'iso' in r