def test_faculty_repo_log_artifacts(mocker, prefix): mocker.patch("faculty.datasets.put") repo = FacultyDatasetsArtifactRepository(ARTIFACT_URI) repo.log_artifacts("/local/dir", prefix + "remote/folder") faculty.datasets.put.assert_called_once_with( "/local/dir", ARTIFACT_ROOT + "remote/folder", PROJECT_ID)
def test_faculty_repo_log_artifacts_default_destination(mocker): mocker.patch("faculty.datasets.put") repo = FacultyDatasetsArtifactRepository(ARTIFACT_URI) repo.log_artifacts("/local/dir") faculty.datasets.put.assert_called_once_with("/local/dir", ARTIFACT_ROOT.rstrip("/"), PROJECT_ID)