mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-29 15:44:18 +01:00
16 lines
449 B
Python
16 lines
449 B
Python
|
import pytest
|
||
|
|
from utils import *
|
||
|
|
|
||
|
|
|
||
|
|
# ref: https://stackoverflow.com/questions/22627659/run-code-before-and-after-each-test-in-py-test
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def stop_server_after_each_test():
|
||
|
|
# do nothing before each test
|
||
|
|
yield
|
||
|
|
# stop all servers after each test
|
||
|
|
instances = set(
|
||
|
|
server_instances
|
||
|
|
) # copy the set to prevent 'Set changed size during iteration'
|
||
|
|
for server in instances:
|
||
|
|
server.stop()
|