def kwargs_matching(): return dict( scheduler=get_versions(), workers={f"worker-{i}": get_versions() for i in range(3)}, client=get_versions(), )
def test_version_custom_pkgs(): out = get_versions([ # Use custom function ("distributed", lambda mod: "123"), # Use version_of_package "notexist", ("pytest", None), # has __version__ "tornado", # has version "math", # has nothing ])["packages"] assert out["distributed"] == "123" assert out["notexist"] is None assert out["pytest"] == pytest.__version__ assert out["tornado"] == tornado.version assert out["math"] is None
import re import sys import pytest import tornado from distributed import Client, Worker from distributed.utils_test import gen_cluster from distributed.versions import error_message, get_versions # if one of the nodes reports this version, there's a mismatch mismatched_version = get_versions() mismatched_version["packages"]["distributed"] = "0.0.0.dev0" # if no key is available for one package, we assume it's MISSING missing_version = get_versions() del missing_version["packages"]["distributed"] # if a node doesn't report any version info, we treat them as UNKNOWN # the happens if the node is pre-32cb96e, i.e. <=2.9.1 unknown_version = None @pytest.fixture def kwargs_matching(): return dict( scheduler=get_versions(), workers={f"worker-{i}": get_versions() for i in range(3)}, client=get_versions(), )
def test_python_version(): required = get_versions()["packages"] assert required["python"] == ".".join(map(str, sys.version_info))
def versions(self, packages=None): return get_versions(packages=packages)