def setUp(self): self.df = pd.DataFrame(np.random.randn(1000, 4), columns=['A', 'B', 'AdmitDTS', 'LastLoadDTS']) self.df['AdmitDTS'] = pd.datetime(2015, 5, 20)