def nyse_dates_advanced(self):
'''
Tests the dates between two dates with the lookBack and lookForward parameters
'''
# Test: with lookbackDays without lookforwardDays
start = datetime(2009, 1, 1)
end = datetime(2011, 1, 1)
dates = DateUtils.nyse_dates(start=start, end=end,
lookbackDays=10, lookforwardDays=0)
self.assertEquals(dates[0], datetime(2008, 12, 17))
self.assertEquals(dates[-1], datetime(2010, 12, 31))
# Test: without lookbackDays with lookforwardDays
start = datetime(2009, 1, 1)
end = datetime(2011, 1, 1)
dates = DateUtils.nyse_dates(start=start, end=end,
lookbackDays=0, lookforwardDays=10)
self.assertEquals(dates[0], datetime(2009, 1, 2))
self.assertEquals(dates[-1], datetime(2011, 1, 14))
# Test: with lookbackDays with lookforwardDays
start = datetime(2009, 1, 1)
end = datetime(2011, 1, 1)
dates = DateUtils.nyse_dates(start=start, end=end,
lookbackDays=10, lookforwardDays=10)
self.assertEquals(dates[0], datetime(2008, 12, 17))
self.assertEquals(dates[-1], datetime(2011, 1, 14))
def get_nyse_dates(self):
# Test: Returns a pd.Series
dates = DateUtils.get_nyse_dates()
self.assertNotEquals(type(dates), list)
self.assertEquals(len(dates), 14728)
# Test: Returns a list
dates = DateUtils.get_nyse_dates(list=True)
self.assertEquals(type(dates), list)
self.assertEquals(len(dates), 14728)
def get_nyse_dates_event(self):
# Test: Returns a pd.Series
dates = DateUtils.get_nyse_dates_event(datetime(2009, 1, 5), 100, 100)
self.assertNotEquals(type(dates), list)
self.assertEquals(len(dates), 201)
self.assertEquals(dates[0], datetime(2008, 8, 12))
self.assertEquals(dates[-1], datetime(2009, 5, 29))
# Test: Returns a list
dates = DateUtils.get_nyse_dates_event(datetime(2009, 1, 5), 50, 50, list=True)
self.assertEquals(type(dates), list)
self.assertEquals(len(dates), 101)
def nyse_add_and_substract(self):
ans = DateUtils.nyse_add(datetime(2009, 4, 13), 5)
self.assertEquals(ans, datetime(2009, 4, 20))
ans = DateUtils.nyse_substract(datetime(2009, 4, 13), 5)
self.assertEquals(ans, datetime(2009, 4, 3))
ans = DateUtils.nyse_add(datetime(1990, 10, 1), 7)
self.assertEquals(ans, datetime(1990, 10, 10))
ans = DateUtils.nyse_substract(datetime(1990, 10, 1), 3)
self.assertEquals(ans, datetime(1990, 9, 26))
def create_trades_from_event(self, eventList,
eventDayAction='Buy', eventDayShares=100,
actionAfter='Sell', daysAfter=5, sharesAfter=100,
actionBefore=None, daysBefore=5, sharesBefore=100):
'''
Creates trades using an event list; usually from the EventFinder.
Also creates aditional order after and before of the event as defined by the user
Parameters
----------
eventList: pandas.Series
'''
self.trades = pd.DataFrame(index=eventList.index, columns=['symbol', 'action', 'num_of_shares'])
self.trades['symbol'] = eventList
self.trades['action'] = eventDayAction
self.trades['num_of_shares'] = eventDayShares
# TODO: Actions BEFORE
if actionAfter is not None:
dicts = []
for idx, row in self.trades.iterrows():
after_date = DateUtils.nyse_add(idx.to_pydatetime(), daysAfter)
after = pd.DataFrame([ {'symbol': row['symbol'],
'action': actionAfter,
'num_of_shares': sharesAfter}],
index=[after_date], columns=self.trades.columns)
self.trades = self.trades.append(after)
self.trades = self.trades.sort()
def create_trades_from_event(self, eventList,
eventDayAction='Buy', eventDayShares=100,
actionAfter='Sell', daysAfter=5, sharesAfter=100,
actionBefore=None, daysBefore=5, sharesBefore=100):
'''
Creates trades using an event list; usually from the EventFinder.
Also creates aditional order after and before of the event as defined by the user
Parameters
----------
eventList: pandas.Series
'''
self.trades = pd.DataFrame(index=eventList.index, columns=['symbol', 'action', 'num_of_shares'])
self.trades['symbol'] = eventList
self.trades['action'] = eventDayAction
self.trades['num_of_shares'] = eventDayShares
# TODO: Actions BEFORE
if actionAfter is not None:
dicts = []
for idx, row in self.trades.iterrows():
after_date = DateUtils.nyse_add(idx.to_pydatetime(), daysAfter)
after = pd.DataFrame([ {'symbol': row['symbol'],
'action': actionAfter,
'num_of_shares': sharesAfter}],
index=[after_date], columns=self.trades.columns)
self.trades = self.trades.append(after)
self.trades = self.trades.sort()
def search(self, oneEventPerEquity=True, useCache=True, save=True):
self.oneEventPerEquity = oneEventPerEquity
# 1. Load the data if requested and available
self.matrix = self.data_access.load(self.generate_filename(),
'.evt_matrix')
if useCache and self.matrix is not None:
pass
else:
# 2. Data was not loaded
# 2.1 Get the dates, and Download/Import the data
nyse_dates = DateUtils.nyse_dates(start=self.start_date,
end=self.end_date)
data = self.data_access.get_data(self.symbols, nyse_dates[0],
nyse_dates[-1], self.field)
# Special case
if len(data.columns) == 1:
data.columns = self.symbols
# 2.2 Create and fill the matrix of events
data = data[self.start_date:self.end_date]
self.matrix = pd.DataFrame(index=data.index, columns=self.symbols)
for symbol in self.symbols:
i = 0
for item in data[symbol][1:]:
e = self.condition.function(i, item, data[symbol][1:])
if e:
self.matrix[symbol][i + 1] = 1
if oneEventPerEquity == True:
break
i = i + 1
# 3. Calculate other results and save if requested
# Reduce Matrix: Sum each row and columns: if is greater than 0 there is an event
self.matrix = self.matrix[self.matrix.fillna(value=0).sum(axis=1) > 0]
valid_cols = self.matrix.columns[self.matrix.fillna(value=0).sum(
axis=0) > 0].values
self.matrix = self.matrix[valid_cols]
# 3.2 Create list of events
self.list = pd.Series(index=self.matrix.index,
name='Equity',
dtype=str)
for idx, row in self.matrix.iterrows():
equity = row[row == 1].index[0]
self.list.ix[idx] = equity
# 3.3 Save
self.num_events = len(self.list)
if save:
self.data_access.save(self.matrix, self.generate_filename(),
'.evt_matrix')
def search(self, oneEventPerEquity=True, useCache=True, save=True):
self.oneEventPerEquity = oneEventPerEquity
# 1. Load the data if requested and available
self.matrix = self.data_access.load(self.generate_filename(), '.evt_matrix')
if useCache and self.matrix is not None:
pass
else:
# 2. Data was not loaded
# 2.1 Get the dates, and Download/Import the data
nyse_dates = DateUtils.nyse_dates(start=self.start_date, end=self.end_date)
data = self.data_access.get_data(self.symbols, nyse_dates[0], nyse_dates[-1], self.field)
# Special case
if len(data.columns) == 1:
data.columns = self.symbols
# 2.2 Create and fill the matrix of events
data = data[self.start_date:self.end_date]
self.matrix = pd.DataFrame(index=data.index, columns=self.symbols)
for symbol in self.symbols:
i = 0
for item in data[symbol][1:]:
e = self.condition.function(i, item, data[symbol][1:])
if e:
self.matrix[symbol][i+1] = 1
if oneEventPerEquity == True:
break
i = i + 1
# 3. Calculate other results and save if requested
# Reduce Matrix: Sum each row and columns: if is greater than 0 there is an event
self.matrix = self.matrix[self.matrix.fillna(value=0).sum(axis=1) > 0]
valid_cols = self.matrix.columns[self.matrix.fillna(value=0).sum(axis=0) > 0].values
self.matrix = self.matrix[valid_cols]
# 3.2 Create list of events
self.list = pd.Series(index=self.matrix.index, name='Equity')
for idx, row in self.matrix.iterrows():
equity = row[row == 1].index[0]
self.list.loc[idx] = equity
# 3.3 Save
self.num_events = len(self.list)
if save:
self.data_access.save(self.matrix, self.generate_filename(), '.evt_matrix')
def run(self):
dates = DateUtils.nyse_dates_event(self.date,
self.lookback_days, self.lookforward_days, self.estimation_period)
start_date = dates[0]
end_date = dates[-1]
# Data to the General market_return Study
self.data = self.data_access.get_data(self.symbol, start_date, end_date, self.field)
evt_window_dates = dates[- self.lookforward_days - self.lookback_days - 1:]
self.evt_window_data = self.data[evt_window_dates[0]:dates[-1]]
self.market = self.data_access.get_data(self.market, start_date, end_date, self.field)
# Parameters of the General market_return Study
self.start_period = dates[0]
self.end_period = dates[self.estimation_period]
self.start_window = dates[self.estimation_period]
self.end_window = dates[-1]
# Run the Market Return method
super().market_return()
def run(self):
dates = DateUtils.get_nyse_dates_event(
self.date, self.lookback_days + self.estimation_period, self.lookforward_days, list=True
)
start_date = dates[0]
end_date = dates[-1]
# Data to the general event
self.data = self.da.get_data(self.symbol, start_date, end_date, self.field)
self.market = self.da.get_data(self.market, start_date, end_date, self.field)
# Parameters of the General Event
self.start_period = dates[0]
self.end_period = dates[self.estimation_period]
self.start_window = dates[self.estimation_period]
self.end_window = dates[-1]
# Run the Market Return method
super().market_return()
def run(self):
dates = DateUtils.nyse_dates_event(self.date, self.lookback_days,
self.lookforward_days,
self.estimation_period)
start_date = dates[0]
end_date = dates[-1]
# Data to the General market_return Study
self.data = self.data_access.get_data(self.symbol, start_date,
end_date, self.field)
evt_window_dates = dates[-self.lookforward_days - self.lookback_days -
1:]
self.evt_window_data = self.data[evt_window_dates[0]:dates[-1]]
self.market = self.data_access.get_data(self.market, start_date,
end_date, self.field)
# Parameters of the General market_return Study
self.start_period = dates[0]
self.end_period = dates[self.estimation_period]
self.start_window = dates[self.estimation_period]
self.end_window = dates[-1]
# Run the Market Return method
super().market_return()
from datetime import datetime from finance.utils import DateUtils all_dates = DateUtils.nyse_dates() print(all_dates) print(DateUtils.nyse_dates(start=datetime(2008,1,1))) index = DateUtils.search_closer_date(datetime(2009,1,1), all_dates) print(index, all_dates[index])
def run(self):
"""
Assess the events
|-----100-----|-------20-------|-|--------20--------|
estimation lookback event lookforward
Prerequisites
-------------
self.matrix
self.market = 'SPY'
self.lookback_days = 20
self.lookforward_days = 20
self.estimation_period = 200
self.field = 'Adj Close'
"""
# 0. Get the dates and Download/Import the data
symbols = list(set(self.list))
start_date = self.list.index[0]
end_date = self.list.index[-1]
nyse_dates = DateUtils.nyse_dates(
start=start_date,
end=end_date,
lookbackDays=self.lookback_days + self.estimation_period + 1,
lookforwardDays=self.lookforward_days,
)
data = self.data_access.get_data(symbols, nyse_dates[0], nyse_dates[-1], self.field)
market = self.data_access.get_data(self.market, nyse_dates[0], nyse_dates[-1], self.field)
if len(data.columns) == 1:
data.columns = symbols
if len(data) > len(market):
market = market.reindex(data.index)
market.columns = [self.field]
data = data.fillna(method="ffill").fillna(method="bfill")
market = market.fillna(method="ffill").fillna(method="bfill")
# 1. Create DataFrames with the data of each event
windows_indexes = range(-self.lookback_days, self.lookforward_days + 1)
estimation_indexes = range(-self.estimation_period - self.lookback_days, -self.lookback_days)
self.equities_window = pd.DataFrame(index=windows_indexes)
self.equities_estimation = pd.DataFrame(index=estimation_indexes)
self.market_window = pd.DataFrame(index=windows_indexes)
self.market_estimation = pd.DataFrame(index=estimation_indexes)
dr_data = Calculator.returns(data)
dr_market = Calculator.returns(market)
self.dr_equities_window = pd.DataFrame(index=windows_indexes)
self.dr_equities_estimation = pd.DataFrame(index=estimation_indexes)
self.dr_market_window = pd.DataFrame(index=windows_indexes)
self.dr_market_estimation = pd.DataFrame(index=estimation_indexes)
# 2. Iterate over the list of events and fill the DataFrames
for i in range(len(self.list)):
symbol = self.list[i]
evt_date = self.list.index[i].to_pydatetime()
col_name = symbol + " " + evt_date.strftime("%Y-%m-%d")
evt_idx = DateUtils.search_closer_date(evt_date, data[symbol].index, exact=True)
# 1.1 Data on the estimation period: self.equities_estimation
start_idx = evt_idx - self.lookback_days - self.estimation_period # estimation start idx on self.data
end_idx = evt_idx - self.lookback_days # estimation end idx on self.data
new_equities_estimation = data[symbol][start_idx:end_idx]
new_equities_estimation.index = self.equities_estimation.index
self.equities_estimation[col_name] = new_equities_estimation
# Daily return of the equities on the estimation period
new_dr_equities_estimation = dr_data[symbol][start_idx:end_idx]
new_dr_equities_estimation.index = self.dr_equities_estimation.index
self.dr_equities_estimation[col_name] = new_dr_equities_estimation
# 1.4 Market on the estimation period: self.market_estimation
new_market_estimation = market[self.field][start_idx:end_idx]
new_market_estimation.index = self.market_estimation.index
self.market_estimation[col_name] = new_market_estimation
# Daily return of the market on the estimation period
new_dr_market_estimation = dr_market[start_idx:end_idx]
new_dr_market_estimation.index = self.dr_market_estimation.index
self.dr_market_estimation[col_name] = new_dr_market_estimation
# 1.3 Equities on the event window: self.equities_window
start_idx = evt_idx - self.lookback_days # window start idx on self.data
end_idx = evt_idx + self.lookforward_days + 1 # window end idx on self.data
new_equities_window = data[symbol][start_idx:end_idx]
new_equities_window.index = self.equities_window.index
self.equities_window[col_name] = new_equities_window
# Daily return of the equities on the event window
new_dr_equities_window = dr_data[symbol][start_idx:end_idx]
new_dr_equities_window.index = self.dr_equities_window.index
self.dr_equities_window[col_name] = new_dr_equities_window
# 1.4 Market on the event window: self.market_window
new_market_window = market[self.field][start_idx:end_idx]
new_market_window.index = self.market_window.index
self.market_window[col_name] = new_market_window
# Daily return of the market on the event window
new_dr_market_window = dr_market[start_idx:end_idx]
new_dr_market_window.index = self.dr_market_window.index
self.dr_market_window[col_name] = new_dr_market_window
# 3. Calculate the linear regression -> expected return
self.reg_estimation = pd.DataFrame(
index=self.dr_market_estimation.columns, columns=["Intercept", "Slope", "Std Error"]
)
self.er = pd.DataFrame(index=self.dr_market_window.index, columns=self.dr_market_window.columns)
# For each column (event) on the estimation period
for col in self.dr_market_estimation.columns:
# 3.1 Calculate the regression
x = self.dr_market_estimation[col]
y = self.dr_equities_estimation[col]
slope, intercept, r_value, p_value, slope_std_error = stats.linregress(x, y)
self.reg_estimation["Slope"][col] = slope
self.reg_estimation["Intercept"][col] = intercept
self.reg_estimation["Std Error"][col] = slope_std_error
# 3.2 Calculate the expected return of each date using the regression
self.er[col] = intercept + self.dr_market_window[col] * slope
# 4. Final results
self.er.columns.name = "Expected return"
self.mean_er = self.er.mean(axis=1)
self.mean_er.name = "Mean ER"
self.std_er = self.er.std(axis=1)
self.std_er.name = "Std ER"
self.ar = self.dr_equities_window - self.er
self.ar.columns.name = "Abnormal return"
self.mean_ar = self.ar.mean(axis=1)
self.mean_ar.name = "Mean AR"
self.std_ar = self.ar.std(axis=1)
self.std_ar.name = "Std AR"
self.car = self.ar.apply(np.cumsum)
self.car.columns.name = "Cum Abnormal Return"
self.mean_car = self.car.mean(axis=1)
self.mean_car.name = "Mean CAR"
self.std_car = self.car.std(axis=1)
self.mean_car.name = "Mean CAR"
def run(self):
'''
Assess the events
|-----100-----|-------20-------|-|--------20--------|
estimation lookback event lookforward
Prerequisites
-------------
self.matrix
self.market = 'SPY'
self.lookback_days = 20
self.lookforward_days = 20
self.estimation_period = 200
self.field = 'Adj Close'
'''
# 0. Get the dates and Download/Import the data
symbols = list(set(self.list))
start_date = self.list.index[0]
end_date = self.list.index[-1]
nyse_dates = DateUtils.nyse_dates(
start=start_date,
end=end_date,
lookbackDays=self.lookback_days + self.estimation_period + 1,
lookforwardDays=self.lookforward_days)
data = self.data_access.get_data(symbols, nyse_dates[0],
nyse_dates[-1], self.field)
market = self.data_access.get_data(self.market, nyse_dates[0],
nyse_dates[-1], self.field)
if len(data.columns) == 1:
data.columns = symbols
if len(data) > len(market):
market = market.reindex(data.index)
market.columns = [self.field]
data = data.fillna(method='ffill').fillna(method='bfill')
market = market.fillna(method='ffill').fillna(method='bfill')
# 1. Create DataFrames with the data of each event
windows_indexes = range(-self.lookback_days, self.lookforward_days + 1)
estimation_indexes = range(
-self.estimation_period - self.lookback_days, -self.lookback_days)
self.equities_window = pd.DataFrame(index=windows_indexes)
self.equities_estimation = pd.DataFrame(index=estimation_indexes)
self.market_window = pd.DataFrame(index=windows_indexes)
self.market_estimation = pd.DataFrame(index=estimation_indexes)
dr_data = Calculator.returns(data)
dr_market = Calculator.returns(market)
self.dr_equities_window = pd.DataFrame(index=windows_indexes)
self.dr_equities_estimation = pd.DataFrame(index=estimation_indexes)
self.dr_market_window = pd.DataFrame(index=windows_indexes)
self.dr_market_estimation = pd.DataFrame(index=estimation_indexes)
# 2. Iterate over the list of events and fill the DataFrames
for i in range(len(self.list)):
symbol = self.list[i]
evt_date = self.list.index[i].to_pydatetime()
col_name = symbol + ' ' + evt_date.strftime('%Y-%m-%d')
evt_idx = DateUtils.search_closer_date(evt_date,
data[symbol].index,
exact=True)
# 1.1 Data on the estimation period: self.equities_estimation
start_idx = evt_idx - self.lookback_days - self.estimation_period # estimation start idx on self.data
end_idx = evt_idx - self.lookback_days # estimation end idx on self.data
new_equities_estimation = data[symbol][start_idx:end_idx]
new_equities_estimation.index = self.equities_estimation.index
self.equities_estimation[col_name] = new_equities_estimation
# Daily return of the equities on the estimation period
new_dr_equities_estimation = dr_data[symbol][start_idx:end_idx]
new_dr_equities_estimation.index = self.dr_equities_estimation.index
self.dr_equities_estimation[col_name] = new_dr_equities_estimation
# 1.4 Market on the estimation period: self.market_estimation
new_market_estimation = market[self.field][start_idx:end_idx]
new_market_estimation.index = self.market_estimation.index
self.market_estimation[col_name] = new_market_estimation
# Daily return of the market on the estimation period
new_dr_market_estimation = dr_market[start_idx:end_idx]
new_dr_market_estimation.index = self.dr_market_estimation.index
self.dr_market_estimation[col_name] = new_dr_market_estimation
# 1.3 Equities on the event window: self.equities_window
start_idx = evt_idx - self.lookback_days # window start idx on self.data
end_idx = evt_idx + self.lookforward_days + 1 # window end idx on self.data
new_equities_window = data[symbol][start_idx:end_idx]
new_equities_window.index = self.equities_window.index
self.equities_window[col_name] = new_equities_window
# Daily return of the equities on the event window
new_dr_equities_window = dr_data[symbol][start_idx:end_idx]
new_dr_equities_window.index = self.dr_equities_window.index
self.dr_equities_window[col_name] = new_dr_equities_window
# 1.4 Market on the event window: self.market_window
new_market_window = market[self.field][start_idx:end_idx]
new_market_window.index = self.market_window.index
self.market_window[col_name] = new_market_window
# Daily return of the market on the event window
new_dr_market_window = dr_market[start_idx:end_idx]
new_dr_market_window.index = self.dr_market_window.index
self.dr_market_window[col_name] = new_dr_market_window
# 3. Calculate the linear regression -> expected return
self.reg_estimation = pd.DataFrame(
index=self.dr_market_estimation.columns,
columns=['Intercept', 'Slope', 'Std Error'])
self.er = pd.DataFrame(index=self.dr_market_window.index,
columns=self.dr_market_window.columns)
# For each column (event) on the estimation period
for col in self.dr_market_estimation.columns:
# 3.1 Calculate the regression
x = self.dr_market_estimation[col]
y = self.dr_equities_estimation[col]
slope, intercept, r_value, p_value, slope_std_error = stats.linregress(
x, y)
self.reg_estimation['Slope'][col] = slope
self.reg_estimation['Intercept'][col] = intercept
self.reg_estimation['Std Error'][col] = slope_std_error
# 3.2 Calculate the expected return of each date using the regression
self.er[col] = intercept + self.dr_market_window[col] * slope
# 4. Final results
self.er.columns.name = 'Expected return'
self.mean_er = self.er.mean(axis=1)
self.mean_er.name = 'Mean ER'
self.std_er = self.er.std(axis=1)
self.std_er.name = 'Std ER'
self.ar = self.dr_equities_window - self.er
self.ar.columns.name = 'Abnormal return'
self.mean_ar = self.ar.mean(axis=1)
self.mean_ar.name = 'Mean AR'
self.std_ar = self.ar.std(axis=1)
self.std_ar.name = 'Std AR'
self.car = self.ar.apply(np.cumsum)
self.car.columns.name = 'Cum Abnormal Return'
self.mean_car = self.car.mean(axis=1)
self.mean_car.name = 'Mean CAR'
self.std_car = self.car.std(axis=1)
self.mean_car.name = 'Mean CAR'
from datetime import datetime from finance.utils import DateUtils all_dates = DateUtils.nyse_dates() print(all_dates) print(DateUtils.nyse_dates(start=datetime(2015,1,1))) index = DateUtils.search_closer_date(datetime(2017,1,1), all_dates) print(index, all_dates[index])
def nyse_dates_event(self):
dates = DateUtils.nyse_dates_event(datetime(2009, 1, 5), 10, 10, 250)
self.assertEquals(dates[0], datetime(2007, 12, 21))
self.assertEquals(dates[-1], datetime(2009, 1, 20))
self.assertEquals(len(dates), 271)
def nyse_dates_basic(self):
'''
Tests the dates without the lookBack and lookForward parameters
'''
# We assume this works: because sometime today is not and open day
today = datetime(datetime.today().year, datetime.today().month, datetime.today().day)
dates = DateUtils.nyse_dates()
today = DateUtils.search_closer_date(today, dates)
today = dates[today]
# End of assumssion
# Test: Returns a list
dates = DateUtils.nyse_dates()
self.assertEquals(type(dates), list)
# Test: Returns a pd.Series if requested
dates = DateUtils.nyse_dates(series=True)
self.assertEquals(type(dates), pd.TimeSeries)
# Test: Default startdate is 2007-1-1
dates = DateUtils.nyse_dates()
ans = dates[0]
self.assertEquals(ans, datetime(2007,1,3))
# Test: Default enddate is today
ans = dates[-1]
self.assertEquals(ans, today)
# Test: Values: start date after 2007-1-1
start = datetime(2008, 1, 1)
# Test: returns list
dates = DateUtils.nyse_dates(start=start)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(2008, 1, 2))
self.assertEquals(dates[-1], today)
# Test: returns pd.Series
dates = DateUtils.nyse_dates(start=start, series=True)
self.assertEquals(type(dates), pd.TimeSeries)
self.assertEquals(dates[0], datetime(2008, 1, 2))
self.assertEquals(dates[-1], today)
# Test: Values: start date before 2007-1-1
start = datetime(1995, 1, 1)
# Test: with list
dates = DateUtils.nyse_dates(start=start)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(1995, 1, 3))
self.assertEquals(dates[-1], today)
# Test: with pd.Series
dates = DateUtils.nyse_dates(start=start, series=True)
self.assertEquals(type(dates), pd.TimeSeries)
self.assertEquals(dates[0], datetime(1995, 1, 3))
self.assertEquals(dates[-1], today)
# Test: end date after 2007-1-1
end = datetime(2009, 6, 6)
dates = DateUtils.nyse_dates(end=end)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(2007, 1, 3))
self.assertEquals(dates[-1], datetime(2009, 6, 5))
# Test: end date before 2007-1-1
end = datetime(2005, 6, 6)
dates = DateUtils.nyse_dates(end=end)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(1962, 7, 5))
self.assertEquals(dates[-1], datetime(2005, 6, 6))
# Test: Values and lenght between 2 dates - No. 1
start = datetime(2000, 1, 1)
end = datetime(2002, 1, 1)
# Test: with list
dates = DateUtils.nyse_dates(start=start, end=end)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(2000, 1, 3))
self.assertEquals(dates[-1], datetime(2001, 12, 31))
self.assertEquals(len(dates), 500)
# Test: with pd.Series
dates = DateUtils.nyse_dates(start=start, end=end, series=True)
self.assertEquals(type(dates), pd.TimeSeries)
self.assertEquals(dates[0], datetime(2000, 1, 3))
self.assertEquals(dates[-1], datetime(2001, 12, 31))
self.assertEquals(len(dates), 500)
# Test: Values and lenght between 2 dates - No. 2
start = datetime(2009, 1, 1)
end = datetime(2011, 1, 1)
# Test: Lenght: Section - list
dates = DateUtils.nyse_dates(start=start, end=end)
self.assertEquals(type(dates), list)
self.assertEquals(dates[0], datetime(2009, 1, 2))
self.assertEquals(dates[-1], datetime(2010, 12, 31))
self.assertEquals(len(dates), 504)
# Test: Lenght: Section - pd.Series
dates = DateUtils.nyse_dates(start=start, end=end, series=True)
self.assertEquals(type(dates), pd.TimeSeries)
self.assertEquals(dates[0], datetime(2009, 1, 2))
self.assertEquals(dates[-1], datetime(2010, 12, 31))
self.assertEquals(len(dates), 504)