chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
import json
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from mlc_llm.support import logging
|
||||
from mlc_llm.support.auto_config import detect_config
|
||||
|
||||
logging.enable_logging()
|
||||
|
||||
# test category "unittest"
|
||||
pytestmark = [pytest.mark.unittest]
|
||||
|
||||
|
||||
def _create_json_file(json_path, data):
|
||||
with open(json_path, "w", encoding="utf-8") as i_f:
|
||||
json.dump(data, i_f)
|
||||
|
||||
|
||||
def test_detect_config():
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
base_path = Path(tmpdir)
|
||||
config_json_path = base_path / "config.json"
|
||||
_create_json_file(config_json_path, {})
|
||||
|
||||
assert detect_config(base_path) == config_json_path
|
||||
assert detect_config(config_json_path) == config_json_path
|
||||
|
||||
|
||||
def test_detect_config_fail():
|
||||
with pytest.raises(ValueError):
|
||||
detect_config(Path("do/not/exist"))
|
||||
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
base_path = Path(tmpdir)
|
||||
with pytest.raises(ValueError):
|
||||
assert detect_config(base_path)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_detect_config()
|
||||
test_detect_config_fail()
|
||||
Reference in New Issue
Block a user