wehub-resource-sync
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
import pytest
|
|
|
|
from mlflow.utils.string_utils import (
|
|
format_table_cell_value,
|
|
is_string_type,
|
|
mslex_quote,
|
|
strip_prefix,
|
|
strip_suffix,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
(("original", "prefix", "expected")),
|
|
[("smoketest", "smoke", "test"), ("", "test", ""), ("", "", ""), ("test", "", "test")],
|
|
)
|
|
def test_strip_prefix(original, prefix, expected):
|
|
assert strip_prefix(original, prefix) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("original", "suffix", "expected"),
|
|
[("smoketest", "test", "smoke"), ("", "test", ""), ("", "", ""), ("test", "", "test")],
|
|
)
|
|
def test_strip_suffix(original, suffix, expected):
|
|
assert strip_suffix(original, suffix) == expected
|
|
|
|
|
|
def test_is_string_type():
|
|
assert is_string_type("validstring")
|
|
assert is_string_type("")
|
|
assert is_string_type((b"dog").decode("utf-8"))
|
|
assert not is_string_type(None)
|
|
assert not is_string_type(["teststring"])
|
|
assert not is_string_type([])
|
|
assert not is_string_type({})
|
|
assert not is_string_type({"test": "string"})
|
|
assert not is_string_type(12)
|
|
assert not is_string_type(12.7)
|
|
|
|
|
|
def test_mslex_quote():
|
|
assert mslex_quote("abc") == "abc"
|
|
assert mslex_quote("a b c") == '"a b c"'
|
|
assert mslex_quote("C:\\path\\to\\file") == "C:\\path\\to\\file"
|
|
|
|
|
|
def test_format_table_cell_value_basic():
|
|
result = format_table_cell_value("field", "value")
|
|
assert result == "value"
|
|
|
|
|
|
def test_format_table_cell_value_empty():
|
|
result = format_table_cell_value("field", None, [])
|
|
assert result == "N/A"
|
|
|
|
|
|
def test_format_table_cell_value_multiple():
|
|
values = ["a", "b", "c", "d", "e"]
|
|
result = format_table_cell_value("field", None, values)
|
|
assert result == "a, b, c, ... (+2 more)"
|
|
|
|
|
|
def test_format_table_cell_value_timestamp():
|
|
timestamp = "2025-01-15T10:31:24.123Z"
|
|
result = format_table_cell_value("info.request_time", timestamp)
|
|
assert "2025-01-15" in result
|
|
assert "10:31:24" in result
|
|
|
|
|
|
def test_format_table_cell_value_duration_ms():
|
|
result = format_table_cell_value("info.execution_duration_ms", 500)
|
|
assert result == "500ms"
|
|
|
|
|
|
def test_format_table_cell_value_duration_seconds():
|
|
result = format_table_cell_value("info.execution_duration_ms", 2500)
|
|
assert result == "2.5s"
|
|
|
|
|
|
def test_format_table_cell_value_preview_truncation():
|
|
long_text = "This is a very long preview text that should be truncated"
|
|
result = format_table_cell_value("info.request_preview", long_text)
|
|
assert result == "This is a very lo..."
|
|
assert len(result) == 20
|
|
|
|
|
|
def test_format_table_cell_value_invalid_timestamp():
|
|
result = format_table_cell_value("info.request_time", "invalid-timestamp")
|
|
assert result == "invalid-timestamp" # Should return original
|
|
|
|
|
|
def test_format_table_cell_value_invalid_duration():
|
|
result = format_table_cell_value("info.execution_duration_ms", "not-a-number")
|
|
assert result == "not-a-number" # Should return original
|