Function test_parametrized

tests/func/test_dataset.py:417–474

(tmp_dir, dvc)

def test_parametrized(tmp_dir, dvc):
    (tmp_dir / "dvc.yaml").dump(
        {
            "datasets": [
                {"name": "${ds1.name}", "url": "${ds1.url}", "type": "dc"},
                {
                    "name": "${ds2.name}",
                    "url": "${ds2.url}",
                    "type": "dvc",
                    "path": "${ds2.path}",
                },
                {
                    "name": "${ds3.name}",
                    "url": "${ds3.url}",
                    "type": "url",
                },
            ]
        }
    )
    (tmp_dir / "params.yaml").dump(
        {
            "ds1": {"name": "dogs", "url": "dc://dogs"},
            "ds2": {
                "name": "example-get-started",
                "url": "git@github.com:iterative/example-get-started.git",
                "path": "path",
            },
            "ds3": {
                "name": "cloud-versioning-demo",
                "url": "s3://cloud-versioning-demo",
            },
        }
    )

    path = (tmp_dir / "dvc.yaml").fs_path
    assert dict(dvc.datasets.items()) == {
        "dogs": DatachainDataset(
            manifest_path=path,
            spec=DatasetSpec(name="dogs", url="dc://dogs", type="dc"),
        ),
        "example-get-started": DVCDataset(
            manifest_path=path,
            spec=DVCDatasetSpec(
                name="example-get-started",
                url="git@github.com:iterative/example-get-started.git",
                path="path",
                type="dvc",
            ),
        ),
        "cloud-versioning-demo": URLDataset(
            manifest_path=path,
            spec=DatasetSpec(
                name="cloud-versioning-demo",
                url="s3://cloud-versioning-demo",
                type="url",
            ),
        ),
    }