|
@@ -509,10 +509,11 @@ class TickingDateTimeFactory:
|
|
|
def __call__(self) -> datetime.datetime:
|
|
|
return self.time_to_freeze + (real_datetime.now() - self.start)
|
|
|
|
|
|
- def tick(self, delta: Union[datetime.timedelta, int]=datetime.timedelta(seconds=1)) -> datetime.datetime:
|
|
|
- if isinstance(delta, numbers.Real):
|
|
|
- # noinspection PyTypeChecker
|
|
|
- self.move_to(self.time_to_freeze + datetime.timedelta(seconds=delta))
|
|
|
+ def tick(self, delta: Union[datetime.timedelta, float]=datetime.timedelta(seconds=1)) -> datetime.datetime:
|
|
|
+ if isinstance(delta, numbers.Integral):
|
|
|
+ self.move_to(self.time_to_freeze + datetime.timedelta(seconds=int(delta)))
|
|
|
+ elif isinstance(delta, numbers.Real):
|
|
|
+ self.move_to(self.time_to_freeze + datetime.timedelta(seconds=float(delta)))
|
|
|
else:
|
|
|
self.move_to(self.time_to_freeze + delta) # type: ignore
|
|
|
return self.time_to_freeze
|
|
@@ -531,10 +532,11 @@ class FrozenDateTimeFactory:
|
|
|
def __call__(self) -> datetime.datetime:
|
|
|
return self.time_to_freeze
|
|
|
|
|
|
- def tick(self, delta: Union[datetime.timedelta, int]=datetime.timedelta(seconds=1)) -> datetime.datetime:
|
|
|
- if isinstance(delta, numbers.Real):
|
|
|
- # noinspection PyTypeChecker
|
|
|
- self.time_to_freeze += datetime.timedelta(seconds=delta)
|
|
|
+ def tick(self, delta: Union[datetime.timedelta, float]=datetime.timedelta(seconds=1)) -> datetime.datetime:
|
|
|
+ if isinstance(delta, numbers.Integral):
|
|
|
+ self.move_to(self.time_to_freeze + datetime.timedelta(seconds=int(delta)))
|
|
|
+ elif isinstance(delta, numbers.Real):
|
|
|
+ self.move_to(self.time_to_freeze + datetime.timedelta(seconds=float(delta)))
|
|
|
else:
|
|
|
self.time_to_freeze += delta # type: ignore
|
|
|
return self.time_to_freeze
|
|
@@ -557,9 +559,13 @@ class StepTickTimeFactory:
|
|
|
self.tick()
|
|
|
return return_time
|
|
|
|
|
|
- def tick(self, delta: Union[datetime.timedelta, int, None]=None) -> datetime.datetime:
|
|
|
+ def tick(self, delta: Union[datetime.timedelta, float, None]=None) -> datetime.datetime:
|
|
|
if not delta:
|
|
|
delta = datetime.timedelta(seconds=self.step_width)
|
|
|
+ elif isinstance(delta, numbers.Integral):
|
|
|
+ delta = datetime.timedelta(seconds=int(delta))
|
|
|
+ elif isinstance(delta, numbers.Real):
|
|
|
+ delta = datetime.timedelta(seconds=float(delta))
|
|
|
self.time_to_freeze += delta # type: ignore
|
|
|
return self.time_to_freeze
|
|
|
|