mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
pre-commit-ci[bot]andSaugat Pachhai
3375a7c39f
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/astral-sh/ruff-pre-commit: v0.3.3 → v0.3.4](https://github.com/astral-sh/ruff-pre-commit/compare/v0.3.3...v0.3.4) * fixes --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Saugat Pachhai (सौगात) <[email protected]>
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
import datetime
|
|
import random
|
|
|
|
import pytest
|
|
|
|
from dvc.repo.experiments.collect import ExpRange, ExpState, SerializableExp, collect
|
|
|
|
|
|
@pytest.mark.vscode
|
|
def test_collect_stable_sorting(dvc, scm, mocker):
|
|
"""
|
|
Check that output is deterministically sorted even for
|
|
commits with the same timestamp. This affects the experience
|
|
in vs-code to avoid experiments "bouncing around" when "exp show"
|
|
is called repeatedly
|
|
"""
|
|
expected_revs = [
|
|
"c" * 40,
|
|
"b" * 40,
|
|
"a" * 40,
|
|
"7" * 40,
|
|
]
|
|
|
|
def collect_queued_patched(_, baseline_revs) -> dict[str, list["ExpRange"]]:
|
|
single_timestamp = datetime.datetime(2023, 6, 20, 0, 0, 0) # noqa: DTZ001
|
|
|
|
exp_ranges = [
|
|
ExpRange(
|
|
revs=[
|
|
ExpState(
|
|
rev=rev,
|
|
name=f"exp-state-{rev[0]}",
|
|
data=SerializableExp(rev=rev, timestamp=single_timestamp),
|
|
)
|
|
],
|
|
name=f"exp-range-{rev[0]}",
|
|
)
|
|
for rev in expected_revs
|
|
]
|
|
|
|
# shuffle collection order
|
|
random.shuffle(exp_ranges)
|
|
|
|
return dict.fromkeys(baseline_revs, exp_ranges)
|
|
|
|
mocker.patch("dvc.repo.experiments.collect.collect_queued", collect_queued_patched)
|
|
mocker.patch("dvc.repo.experiments.collect.collect_active", return_value={})
|
|
mocker.patch("dvc.repo.experiments.collect.collect_failed", return_value={})
|
|
mocker.patch("dvc.repo.experiments.collect.collect_successful", return_value={})
|
|
|
|
# repeat (shuffling collection order in collect_queued_patched)
|
|
for _ in range(20):
|
|
collected = collect(repo=dvc, all_commits=True)
|
|
assert collected[0].rev == "workspace"
|
|
assert collected[0].experiments is None
|
|
assert collected[1].rev == scm.get_rev()
|
|
_assert_experiment_rev_order(collected[1].experiments, expected_revs)
|
|
|
|
|
|
def _assert_experiment_rev_order(actual: list["ExpRange"], expected_revs: list[str]):
|
|
expected_revs = expected_revs.copy()
|
|
|
|
for actual_exp_range in actual:
|
|
for exp_state in actual_exp_range.revs:
|
|
assert exp_state.rev == expected_revs.pop(0)
|