def test_use_energy(): energy = Energy(10, 1000) assert energy == 10 energy.use() assert energy == 9 energy.use(5) assert energy == 4 with raises(ValueError): energy.use(5) energy.use(10, timestamp() + 10000) with raises(ValueError): energy.use(10, timestamp() + 10010)
def test_use_energy(): energy = Energy(10, 1000) assert energy == 10 energy.use() assert energy == 9 energy.use(5) assert energy == 4 with raises(ValueError): energy.use(5) energy.use(10, timestamp() + 10000) assert energy.current(timestamp() + 10000) == 0 with raises(ValueError): energy.use(10, timestamp() + 10010) energy.use(10, timestamp() + 10010, force=True) assert energy.recover_in(timestamp() + 10010) == 11000 assert energy.current(timestamp() + 10010) == 0 assert energy.debt(timestamp() + 10010) == 10 energy.use(10, timestamp() + 10010, force=True) assert energy.recover_in(timestamp() + 10010) == 21000 assert energy.debt(timestamp() + 10010) == 20
def test_timestamp(): _1sec = timedelta(0, 1) assert datetime.utcfromtimestamp(timestamp()) - datetime.utcnow() < _1sec assert datetime.fromtimestamp(timestamp()) - datetime.now() < _1sec assert timestamp() - timegm(gmtime()) < 1