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)