def test_exiting_context_manager_of_cloud_dataset_updates_cloud_metadata(
self):
"""Test that cloud metadata for a cloud dataset is updated on exit of the dataset context manager."""
dataset_path = self._create_nested_cloud_dataset()
dataset = Dataset(path=dataset_path, recursive=True)
with dataset:
dataset.tags = {"cat": "dog"}
dataset.labels = {"animals"}
reloaded_dataset = Dataset(path=dataset_path)
self.assertEqual(reloaded_dataset.id, dataset.id)
self.assertEqual(reloaded_dataset.tags, {"cat": "dog"})
self.assertEqual(reloaded_dataset.labels, {"animals"})
def test_exiting_context_manager_of_local_dataset_updates_local_metadata(
self):
"""Test that local metadata for a local dataset is updated on exit of the dataset context manager."""
with tempfile.TemporaryDirectory() as temporary_directory:
self._create_files_and_nested_subdirectories(temporary_directory)
dataset = Dataset(path=temporary_directory, recursive=True)
with dataset:
dataset.tags = {"cat": "dog"}
dataset.labels = {"animals"}
reloaded_dataset = Dataset(path=temporary_directory,
recursive=True)
self.assertEqual(reloaded_dataset.id, dataset.id)
self.assertEqual(reloaded_dataset.tags, {"cat": "dog"})
self.assertEqual(reloaded_dataset.labels, {"animals"})
