def test_record_incr(self): algo = RecordAlgorithm(sim_params=self.sim_params, data_frequency='daily') output = algo.run(self.source) np.testing.assert_array_equal(output['incr'].values, range(1, len(output) + 1))
def test_record_incr(self): algo = RecordAlgorithm( sim_params=self.sim_params) output = algo.run(self.source) np.testing.assert_array_equal(output['incr'].values, range(1, len(output) + 1)) np.testing.assert_array_equal(output['name'].values, range(1, len(output) + 1)) np.testing.assert_array_equal(output['name2'].values, [2] * len(output)) np.testing.assert_array_equal(output['name3'].values, range(1, len(output) + 1))
def test_record_incr(self): algo = RecordAlgorithm() output = algo.run(self.source) np.testing.assert_array_equal(output['incr'].values, range(1, len(output) + 1))