예제 #1
0
class TestPriceFunctions():
    def setup_method(self, test_method):
        self.pf = PriceFunctions()
        self.coins = ['BTC', 'ETH', 'DASH', 'DOGE', 'LTC', 'STR', 'XMR', 'XRP']

    def test_to_same_starting(self):
        self.coins = ['BTC', 'ETH', 'DASH', 'DOGE', 'LTC', 'STR', 'XMR', 'XRP']

        dfs = {}

        for coin in self.coins:
            dfs[coin] = self.pf.get_pandas(coin=coin,
                                           targetdays=24,
                                           absolute=True)

        dfs = self.pf.to_same_starting(dfs)

        assert (sum(dfs['BTC'].index != dfs['ETH'].index) == 0)
        assert (sum(dfs['ETH'].index != dfs['DASH'].index) == 0)
        assert (sum(dfs['DASH'].index != dfs['DOGE'].index) == 0)
        assert (sum(dfs['DOGE'].index != dfs['LTC'].index) == 0)
        assert (sum(dfs['LTC'].index != dfs['STR'].index) == 0)
        assert (sum(dfs['STR'].index != dfs['XMR'].index) == 0)
        assert (sum(dfs['XMR'].index != dfs['XRP'].index) == 0)

    def test_get_pandas(self):

        for coin in self.coins:
            df = PriceFunctions().get_pandas(coin=coin, targetdays=24)
            assert (sum(df.index[1:] - df.index[:-1] != 3600) == 0)
            assert ({
                'Open', 'Open', 'High', 'Low', 'Volume', 'Percentage Change',
                'Classification'
            }.issubset(df.columns))
    def test_mergeData(self):
        pf = PriceFunctions()
        self.btc = pf.get_pandas(coin='BTC', targetdays=24,
                                 absolute=True)  #as the data is hourly
        self.btc = self.btc.reset_index()

        xmr = pd.read_csv('CryptoScraper/cache/XMR.csv')
        dash = pd.read_csv('CryptoScraper/cache/DASH.csv')
        doge = pd.read_csv('CryptoScraper/cache/DOGE.csv')
        eth = pd.read_csv('CryptoScraper/cache/ETH.csv')
        ltc = pd.read_csv('CryptoScraper/cache/LTC.csv')
        str = pd.read_csv('CryptoScraper/cache/STR.csv')
        xrp = pd.read_csv('CryptoScraper/cache/XRP.csv')

        self.btcXmr = self.btc[self.btc['Date'] >= xmr.iloc[0]['Date']]
        self.btcXmr.reset_index(inplace=True, drop=True)

        assert (sum(self.btcXmr['Date'] != xmr['Date']) == 0)

        self.btcDash = self.btc[self.btc['Date'] >= dash.iloc[0]['Date']]
        self.btcDash.reset_index(inplace=True, drop=True)

        assert (sum(self.btcDash['Date'] != dash['Date']) == 0)

        self.btcDoge = self.btc[self.btc['Date'] >= doge.iloc[0]['Date']]
        self.btcDoge.reset_index(inplace=True, drop=True)

        assert (sum(self.btcDoge['Date'] != doge['Date']) == 0)

        self.btcEth = self.btc[self.btc['Date'] >= eth.iloc[0]['Date']]
        self.btcEth.reset_index(inplace=True, drop=True)

        assert (sum(self.btcEth['Date'] != eth['Date']) == 0)

        self.btcLtc = self.btc[self.btc['Date'] >= ltc.iloc[0]['Date']]
        self.btcLtc.reset_index(inplace=True, drop=True)

        assert (sum(self.btcLtc['Date'] != ltc['Date']) == 0)

        self.btcStr = self.btc[self.btc['Date'] >= str.iloc[0]['Date']]
        self.btcStr.reset_index(inplace=True, drop=True)

        assert (sum(self.btcStr['Date'] != str['Date']) == 0)

        self.btcXrp = self.btc[self.btc['Date'] >= xrp.iloc[0]['Date']]
        self.btcXrp.reset_index(inplace=True, drop=True)

        assert (sum(self.btcXrp['Date'] != xrp['Date']) == 0)
예제 #3
0
    def test_get_pandas(self):

        for coin in self.coins:
            df = PriceFunctions().get_pandas(coin=coin, targetdays=24)
            assert (sum(df.index[1:] - df.index[:-1] != 3600) == 0)
            assert ({
                'Open', 'Open', 'High', 'Low', 'Volume', 'Percentage Change',
                'Classification'
            }.issubset(df.columns))
예제 #4
0
 def setup_method(self, test_method):
     self.pf = PriceFunctions()
     self.coins = ['BTC', 'ETH', 'DASH', 'DOGE', 'LTC', 'STR', 'XMR', 'XRP']
예제 #5
0
 def __init__(self, symbol):
     pf = PriceFunctions()
     self.btc = pf.get_pandas(coin='BTC', targetdays=24, absolute=True) #as the data is hourly
     self.btc = self.btc.reset_index()
     self.url = "https://poloniex.com/public?command=returnChartData&currencyPair=BTC_{}&start={}&end=9999999999&period={}" #gives half an hour data
     self.symbol = symbol