def test_update(): model = BmiHeat() model.initialize() for inc in xrange(10): model.update() assert_almost_equal(model.get_current_time(), (inc + 1) * model.get_time_step())