/ tests / utils / test_time.py
test_time.py
 1  import time
 2  
 3  from mlflow.utils.time import Timer
 4  
 5  
 6  def test_timer():
 7      with Timer() as t:
 8          time.sleep(0.1)
 9  
10      assert f"{t}" == f"{t.elapsed}"
11      assert f"{t:.3f}" == f"{t.elapsed:.3f}"