python-notes/freezegun_tutorial.py at master · twtrubiks/python-notes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# freezegun (https://github.com/spulec/freezegun)
# pip3 install freezegun

from freezegun import freeze_time
from datetime import datetime
import time

# Decorator
@freeze_time("2021-10-10")
def example_1():
    assert datetime.now() == datetime(2021, 10, 10)

# Context manager
def example_2():
    assert datetime.now() != datetime(2021, 10, 10)
    with freeze_time("2021-10-10"):
        assert datetime.now() == datetime(2021, 10, 10)
    assert datetime.now() != datetime(2021, 10, 10)

# Raw use
def example_3():
    freezer = freeze_time("2021-10-10 10:00:00")
    freezer.start()
    assert datetime.now() == datetime(2021, 10, 10, 10, 0, 0)
    freezer.stop()
    assert datetime.now() != datetime(2021, 10, 10, 10, 0, 0)

# tick argument
@freeze_time("2021-10-10", tick=True)
def example_4():
    print(datetime.now()) # 2021-10-10 00:00:00
    time.sleep(5)
    assert datetime.now() > datetime(2021, 10, 10)
    print(datetime.now()) # 2021-10-10 00:00:05

# auto_tick_seconds
@freeze_time("2021-10-10", auto_tick_seconds=15)
def example_5():
    print(datetime.now()) # 2021-10-10 00:00:00
    print(datetime.now()) # 2021-10-10 00:00:15

if __name__ == '__main__':
    example_1()
    # example_2()
    # example_3()
    # example_4()
    # example_5()