fix tests for author

Author: jniestroy

tests/test_dataset.py

@@ -14,8 +14,8 @@ def test_dataset_instantiation(dataset_minimal_data):
     assert dataset.wasGeneratedBy == []  # No generatedBy provided
     assert dataset.wasDerivedFrom == []  # No derivedFrom provided
     assert len(dataset.wasAttributedTo) == 1
-    assert isinstance(dataset.wasAttributedTo[0], IdentifierValue)
-    assert dataset.wasAttributedTo[0].guid == dataset_minimal_data["author"]
+    assert isinstance(dataset.wasAttributedTo[0], str)
+    assert dataset.wasAttributedTo[0] == dataset_minimal_data["author"]
 
 def test_dataset_missing_required_field(dataset_minimal_data):
     """Test that a ValidationError is raised for a missing required field."""
@@ -67,8 +67,8 @@ def test_dataset_with_provenance(dataset_minimal_data):
     assert dataset.wasDerivedFrom[0].guid == "ark:59852/dataset-source"
 
     assert len(dataset.wasAttributedTo) == 2
-    assert all(isinstance(item, IdentifierValue) for item in dataset.wasAttributedTo)
-    author_ids = [item.guid for item in dataset.wasAttributedTo]
+    assert all(isinstance(item, str) for item in dataset.wasAttributedTo)
+    author_ids = [item for item in dataset.wasAttributedTo]
     assert "Author 1" in author_ids
     assert "Author 2" in author_ids
 

tests/test_mlmodel.py

@@ -22,8 +22,8 @@ def test_mlmodel_instantiation(mlmodel_minimal_data):
 
     # Test PROV field auto-population
     assert len(mlmodel.wasAttributedTo) == 1
-    assert isinstance(mlmodel.wasAttributedTo[0], IdentifierValue)
-    assert mlmodel.wasAttributedTo[0].guid == mlmodel_minimal_data["author"]
+    assert isinstance(mlmodel.wasAttributedTo[0], str)
+    assert mlmodel.wasAttributedTo[0] == mlmodel_minimal_data["author"]
 
 def test_mlmodel_missing_required_field(mlmodel_minimal_data):
     """Test ValidationError for missing a required field."""
@@ -45,8 +45,8 @@ def test_mlmodel_with_multiple_authors(mlmodel_minimal_data):
 
     # Test PROV:wasAttributedTo handles list of authors
     assert len(mlmodel.wasAttributedTo) == 2
-    assert all(isinstance(item, IdentifierValue) for item in mlmodel.wasAttributedTo)
-    author_ids = [item.guid for item in mlmodel.wasAttributedTo]
+    assert all(isinstance(item, str) for item in mlmodel.wasAttributedTo)
+    author_ids = [item for item in mlmodel.wasAttributedTo]
     assert "ML Author 1" in author_ids
     assert "ML Author 2" in author_ids