@@ -0,0 +1,1389 @@
+from nose.tools import assert_equal%0A%0Afrom pymt.framework.bmi_bridge import _BmiCap%0A%0A%0Aclass SimpleTimeBmi():%0A    def get_time_units(self):%0A        return 0, 'h'%0A%0A    def get_start_time(self):%0A        return 0, 1.%0A%0A    def get_current_time(self):%0A        return 0, 10.5%0A%0A    def get_end_time(self):%0A        return 0, 72%0A%0A    def get_time_step(self):%0A        return 0, 0.25%0A%0A%0Aclass Bmi(_BmiCap):%0A    _cls = SimpleTimeBmi%0A%0A%0Adef test_time_wrap():%0A    bmi = Bmi()%0A%0A    assert_equal(bmi.get_time_units(), 'h')%0A    assert_equal(bmi.get_start_time(), 1.)%0A    assert_equal(bmi.get_current_time(), 10.5)%0A    assert_equal(bmi.get_end_time(), 72.)%0A    assert_equal(bmi.get_time_step(), .25)%0A    assert_equal(bmi.time_units, 'h')%0A%0A%0Adef test_time_conversion():%0A    bmi = Bmi()%0A%0A    assert_equal(bmi.get_start_time(units='h'), 1.)%0A%0A    assert_equal(bmi.get_start_time(units='min'), 60.)%0A    assert_equal(bmi.get_current_time(units='min'), 630.)%0A    assert_equal(bmi.get_end_time(units='d'), 3)%0A%0A%0Adef test_change_time_units():%0A    bmi = Bmi()%0A%0A    assert_equal(bmi.time_units, 'h')%0A    bmi.time_units = 'min'%0A%0A    assert_equal(bmi.get_start_time(), 60.)%0A    assert_equal(bmi.get_current_time(), 630.)%0A    assert_equal(bmi.get_end_time(), 72 * 60)%0A%0A    assert_equal(bmi.get_start_time(units='h'), 1.)%0A    assert_equal(bmi.get_current_time(units='h'), 10.5)%0A    assert_equal(bmi.get_end_time(units='h'), 72)%0A

Add tests for time units methods.