def test_generate(self): rng1 = list(generate_range(START, END, offset=datetools.bday)) rng2 = list(generate_range(START, END, time_rule='WEEKDAY')) self.assert_(np.array_equal(rng1, rng2))
def eqXDateRange(kwargs, expected): rng = generate_range(**kwargs) assert(np.array_equal(list(rng), expected))