def test_track_the_best():
main_loop = MockMainLoop()
extension = TrackTheBest("cost")
extension.main_loop = main_loop
main_loop.status['iterations_done'] += 1
main_loop.log.current_row['cost'] = 5
extension.dispatch('after_batch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row['cost_best_so_far']
main_loop.status['iterations_done'] += 1
main_loop.log.current_row['cost'] = 6
extension.dispatch('after_batch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row.get('cost_best_so_far', None) is None
main_loop.status['iterations_done'] += 1
main_loop.log.current_row['cost'] = 5
extension.dispatch('after_batch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row.get('cost_best_so_far', None) is None
main_loop.status['iterations_done'] += 1
main_loop.log.current_row['cost'] = 4
extension.dispatch('after_batch')
assert main_loop.status['best_cost'] == 4
assert main_loop.log.current_row['cost_best_so_far']
def test_track_the_best():
main_loop = MockMainLoop()
extension = TrackTheBest("cost")
extension.main_loop = main_loop
main_loop.status['epochs_done'] += 1
main_loop.status['iterations_done'] += 10
main_loop.log.current_row['cost'] = 5
extension.dispatch('after_epoch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row['cost_best_so_far']
main_loop.status['epochs_done'] += 1
main_loop.status['iterations_done'] += 10
main_loop.log.current_row['cost'] = 6
extension.dispatch('after_epoch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row.get('cost_best_so_far', None) is None
main_loop.status['epochs_done'] += 1
main_loop.status['iterations_done'] += 10
main_loop.log.current_row['cost'] = 5
extension.dispatch('after_epoch')
assert main_loop.status['best_cost'] == 5
assert main_loop.log.current_row.get('cost_best_so_far', None) is None
main_loop.status['epochs_done'] += 1
main_loop.status['iterations_done'] += 10
main_loop.log.current_row['cost'] = 4
extension.dispatch('after_epoch')
assert main_loop.status['best_cost'] == 4
assert main_loop.log.current_row['cost_best_so_far']
def test_track_the_best():
main_loop = MockMainLoop()
extension = TrackTheBest("cost")
extension.main_loop = main_loop
main_loop.status["epochs_done"] += 1
main_loop.status["iterations_done"] += 10
main_loop.log.current_row["cost"] = 5
extension.dispatch("after_epoch")
assert main_loop.status["best_cost"] == 5
assert main_loop.log.current_row["cost_best_so_far"]
main_loop.status["epochs_done"] += 1
main_loop.status["iterations_done"] += 10
main_loop.log.current_row["cost"] = 6
extension.dispatch("after_epoch")
assert main_loop.status["best_cost"] == 5
assert main_loop.log.current_row.get("cost_best_so_far", None) is None
main_loop.status["epochs_done"] += 1
main_loop.status["iterations_done"] += 10
main_loop.log.current_row["cost"] = 5
extension.dispatch("after_epoch")
assert main_loop.status["best_cost"] == 5
assert main_loop.log.current_row.get("cost_best_so_far", None) is None
main_loop.status["epochs_done"] += 1
main_loop.status["iterations_done"] += 10
main_loop.log.current_row["cost"] = 4
extension.dispatch("after_epoch")
assert main_loop.status["best_cost"] == 4
assert main_loop.log.current_row["cost_best_so_far"]
def test_track_the_best():
main_loop = MockMainLoop()
extension = TrackTheBest("cost")
extension.main_loop = main_loop
main_loop.status.iterations_done += 1
main_loop.log.current_row.cost = 5
extension.dispatch('after_batch')
assert main_loop.status.best_cost == 5
assert main_loop.log.current_row['cost_best_so_far']
main_loop.status.iterations_done += 1
main_loop.log.current_row.cost = 6
extension.dispatch('after_batch')
assert main_loop.status.best_cost == 5
assert main_loop.log.current_row['cost_best_so_far'] is None
main_loop.status.iterations_done += 1
main_loop.log.current_row.cost = 5
extension.dispatch('after_batch')
assert main_loop.status.best_cost == 5
assert main_loop.log.current_row['cost_best_so_far'] is None
main_loop.status.iterations_done += 1
main_loop.log.current_row.cost = 4
extension.dispatch('after_batch')
assert main_loop.status.best_cost == 4
assert main_loop.log.current_row['cost_best_so_far']
