wehub-resource-sync
14 lines
427 B
Python
14 lines
427 B
Python
from mlflow.tracking.request_header.default_request_header_provider import (
|
|
_DEFAULT_HEADERS,
|
|
DefaultRequestHeaderProvider,
|
|
)
|
|
|
|
|
|
def test_default_request_header_provider_in_context():
|
|
assert DefaultRequestHeaderProvider().in_context()
|
|
|
|
|
|
def test_default_request_header_provider_request_headers():
|
|
request_headers = DefaultRequestHeaderProvider().request_headers()
|
|
assert request_headers == _DEFAULT_HEADERS
|