def bonus_energy(): energy = Energy(10, 300) with time_traveler() as T: T(0) energy.set(15) T(1) assert energy == 15 T(2) energy.use() T(3) assert energy.recover_in() is None T(4) energy.use() T(5) assert energy.recover_in() is None T(6) energy.use(5) T(7) assert energy.recover_in() == 299 T(8) assert energy.recover_in() == 298 T(9) energy.set(15) T(10) assert energy.recover_in() is None
def test_extra_energy(): energy = Energy(10, 300) with time_traveler() as T: T(0) energy.set(15) T(1) assert energy == 15 assert energy.recover_in() is None assert energy.recover_fully_in() is None T(2) energy.use() assert energy.recover_in() is None assert energy.recover_fully_in() is None T(6) energy.use(6) T(7) assert energy.recover_in() == 299 assert energy.recover_fully_in() == 599 T(8) assert energy.recover_in() == 298 assert energy.recover_fully_in() == 598 T(9) energy.set(15) assert energy.recover_in() is None assert energy.recover_fully_in() is None T(10) assert energy.recover_in() is None assert energy.recover_fully_in() is None
def bonus_energy(): energy = Energy(10, 300) with time_traveler() as T: T( 0); energy.set(15) T( 1); assert energy == 15 T( 2); energy.use() T( 3); assert energy.recover_in() is None T( 4); energy.use() T( 5); assert energy.recover_in() is None T( 6); energy.use(5) T( 7); assert energy.recover_in() == 299 T( 8); assert energy.recover_in() == 298 T( 9); energy.set(15) T(10); assert energy.recover_in() is None
def set_energy(): energy = Energy(10, 1000) energy.set(1) assert energy == 1 energy.set(5) assert energy == 5
def test_set_energy(): energy = Energy(10, 1000) energy.set(1) assert energy == 1 energy.set(5) assert energy == 5