mirror of
https://github.com/microsoft/graphrag.git
synced 2025-08-16 12:41:25 +00:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
# Copyright (c) 2024 Microsoft Corporation.
|
||
|
|
# Licensed under the MIT License
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from graphrag.config.resolve_timestamp_path import resolve_timestamp_path
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_timestamp_path_no_timestamp_with_run_id():
|
||
|
|
path = Path("path/to/data")
|
||
|
|
result = resolve_timestamp_path(path, "20240812-121000")
|
||
|
|
assert result == path
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_timestamp_path_no_timestamp_without_run_id():
|
||
|
|
path = Path("path/to/data")
|
||
|
|
result = resolve_timestamp_path(path)
|
||
|
|
assert result == path
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_timestamp_path_with_timestamp_and_run_id():
|
||
|
|
path = Path("some/path/${timestamp}/data")
|
||
|
|
expected = Path("some/path/20240812/data")
|
||
|
|
result = resolve_timestamp_path(path, "20240812")
|
||
|
|
assert result == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_timestamp_path_with_timestamp_and_inferred_directory():
|
||
|
|
cwd = Path(__file__).parent
|
||
|
|
path = cwd / "fixtures/timestamp_dirs/${timestamp}/data"
|
||
|
|
expected = cwd / "fixtures/timestamp_dirs/20240812-120000/data"
|
||
|
|
result = resolve_timestamp_path(path)
|
||
|
|
assert result == expected
|