Files
dvc/tests/utils/asserts.py
T
skshetry a137b2a6bc update ruff and enable flake8-logging-format and flake8-type-checking (#8894)
* update ruff and enable flake8-logging-format and flake8-type-checking check

* Update dvc/external_repo.py

* Update dvc/fs/dvc.py

* Update dvc/repo/experiments/__init__.py

* Update dvc/repo/experiments/queue/base.py

* Update pyproject.toml
2023-01-27 19:14:19 +05:45

22 lines
593 B
Python

from typing import TYPE_CHECKING, Any, Dict
from unittest.mock import ANY
if TYPE_CHECKING:
from unittest.mock import Mock
def issubset(subset: Dict, superset: Dict) -> bool:
assert {**superset, **subset} == superset
return True
def called_once_with_subset(m: "Mock", *args: Any, **kwargs: Any) -> bool:
m.assert_called_once()
m_args, m_kwargs = m.call_args
expected_args = m_args + (ANY,) * (len(m_args) - len(args) - 1)
expected_kwargs = {k: kwargs.get(k, ANY) for k in m_kwargs}
m.assert_called_with(*expected_args, **expected_kwargs)
return True