Module `xchg.tests.test_download_candles`

Unit tests for download_candles.py.

Expand source code

'''Unit tests for download_candles.py.'''

import os
import pandas as pd
from ..download_candles import _request
from ..download_candles import _candles_to_df
from ..download_candles import _main


def test_candles_to_df(downloaded_candles: list,
                       dataframe: pd.core.frame.DataFrame):
    '''Test conversion from candles to a Pandas DataFrame.

    Args:
        downloaded_candles: Test candles in a list.
        dataframe: Pandas DataFrame with test candles.
    '''
    downloaded_df = _candles_to_df(downloaded_candles)
    pd.testing.assert_frame_equal(downloaded_df, dataframe,
                                  check_exact=True)


def test_request(downloaded_candles: list):
    '''Test request to Poloniex.

    Args:
        downloaded_candles: Test candles in a list.
    '''
    candles = _request('ETH', 1800, 1575158400, 1575160200)
    assert candles == downloaded_candles


def test_main(tmp_path: str):
    '''Test main function.

    Args:
        tmp_path: Path which authomatically created by pytest for testing.
    '''
    path = tmp_path / 'sample_data'
    _main(data_folder=path)
    currencies = {'ETH.csv', 'ETC.csv', 'XMR.csv', 'LTC.csv'}
    assert set(os.listdir(path)) == currencies

Functions

def test_candles_to_df(downloaded_candles: list, dataframe: pandas.core.frame.DataFrame)

Test conversion from candles to a Pandas DataFrame.

Args

downloaded_candles: Test candles in a list.
dataframe: Pandas DataFrame with test candles.

Expand source code

def test_candles_to_df(downloaded_candles: list,
                       dataframe: pd.core.frame.DataFrame):
    '''Test conversion from candles to a Pandas DataFrame.

    Args:
        downloaded_candles: Test candles in a list.
        dataframe: Pandas DataFrame with test candles.
    '''
    downloaded_df = _candles_to_df(downloaded_candles)
    pd.testing.assert_frame_equal(downloaded_df, dataframe,
                                  check_exact=True)

def test_main(tmp_path: str)

Test main function.

Args

tmp_path: Path which authomatically created by pytest for testing.

Expand source code

def test_main(tmp_path: str):
    '''Test main function.

    Args:
        tmp_path: Path which authomatically created by pytest for testing.
    '''
    path = tmp_path / 'sample_data'
    _main(data_folder=path)
    currencies = {'ETH.csv', 'ETC.csv', 'XMR.csv', 'LTC.csv'}
    assert set(os.listdir(path)) == currencies

def test_request(downloaded_candles: list)

Test request to Poloniex.

Args

downloaded_candles: Test candles in a list.

Expand source code

def test_request(downloaded_candles: list):
    '''Test request to Poloniex.

    Args:
        downloaded_candles: Test candles in a list.
    '''
    candles = _request('ETH', 1800, 1575158400, 1575160200)
    assert candles == downloaded_candles