def test__setup_engine():
engine = Engine(lambda e, b: 1)
engine.state = State(iteration=10, epoch=1, max_epochs=100, epoch_length=100)
data = list(range(100))
engine.state.dataloader = data
engine._setup_engine()
assert len(engine._init_iter) == 1 and engine._init_iter[0] == 10
def test__setup_engine():
engine = Engine(lambda e, b: 1)
engine.state = State(
iteration=10, epoch=1, max_epochs=100, epoch_length=100, seed=12
)
data = list(range(100))
engine.state.dataloader = data
engine._setup_engine()
assert engine._dataloader_len == len(data)
