def test_double_register_dataset(env: CompilerEnv): dataset = make_dataset() assert env.register_dataset(dataset) with pytest.raises(ValueError) as ctx: env.register_dataset(dataset) assert str( ctx.value) == f"Dataset already registered with name: {dataset.name}"
def test_register_dataset_different_platform(env: CompilerEnv): dataset = make_dataset(platforms=["not-a-real-platform"]) assert not env.register_dataset(dataset) assert dataset.name not in env.available_datasets
def test_register_dataset_matching_platform(env: CompilerEnv): platform = {"darwin": "macos"}.get(sys.platform, sys.platform) dataset = make_dataset(platforms=[platform]) assert env.register_dataset(dataset) assert dataset.name in env.available_datasets
def test_register_dataset(env: CompilerEnv): dataset = make_dataset() assert env.register_dataset(dataset) assert dataset.name in env.available_datasets