50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
# coding=utf-8
|
|
from unittest import TestCase
|
|
from datetime import datetime
|
|
|
|
from metrik.tasks.state_street import StateStreetHoldings
|
|
|
|
|
|
class StateStreetHoldingTest(TestCase):
|
|
def test_spy_holdings(self):
|
|
holdings_dict = StateStreetHoldings.retrieve_data(
|
|
'SPY', datetime.now(), True
|
|
)
|
|
|
|
self.assertEqual(holdings_dict['Ticker Symbol'], 'SPY')
|
|
self.assertEqual(holdings_dict['Fund Name'], u'SPDR® S&P 500® ETF')
|
|
self.assertGreaterEqual(len(holdings_dict['holdings']), 500)
|
|
# Long live AAPL
|
|
self.assertTrue(holdings_dict['holdings'][0]['Identifier'] == u'AAPL')
|
|
|
|
def test_sdy_holdings(self):
|
|
holdings_dict = StateStreetHoldings.retrieve_data(
|
|
'SDY', datetime.now(), True
|
|
)
|
|
|
|
self.assertEqual(holdings_dict['Ticker Symbol'], 'SDY')
|
|
self.assertEqual(holdings_dict['Fund Name'], u'SPDR® S&P® Dividend ETF')
|
|
self.assertEqual(holdings_dict['holdings'][0]['Identifier'], 'T')
|
|
|
|
def test_spyd_holdings(self):
|
|
holdings_dict = StateStreetHoldings.retrieve_data(
|
|
'SPYD', datetime.now(), True
|
|
)
|
|
|
|
self.assertEqual(holdings_dict['Ticker Symbol'], 'SPYD')
|
|
self.assertEqual(holdings_dict['Fund Name'], u'SPDR® S&P® 500 High Dividend ETF')
|
|
|
|
def test_r3k_holdings(self):
|
|
holdings_dict = StateStreetHoldings.retrieve_data(
|
|
'THRK', datetime.now(), True
|
|
)
|
|
|
|
self.assertEqual(holdings_dict['Ticker Symbol'], 'THRK')
|
|
self.assertEqual(holdings_dict['Fund Name'], u'SPDR Russell 3000® ETF')
|
|
# Interesting story: the fund is not required to actually invest in all
|
|
# 3000 Russell equities, but just seeks to track the index in general.
|
|
# That's why the test is against 2000, not 3000.
|
|
# This also means that we can't check lists of say iShares against this
|
|
# because they're not guaranteed to be consistent.
|
|
self.assertGreaterEqual(len(holdings_dict['holdings']), 2000)
|