File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed
Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -221,7 +221,8 @@ def __str__(self):
221221
222222 def get (self , key : str , default : Any = None ) -> Any :
223223 """Get value for a given metadata key."""
224- return self .__dict__ .get (key , default )
224+ value = self .__dict__ .get (key )
225+ return default if value is None else value
225226
226227 def pop (self , key : str , default : Any = None ) -> Any :
227228 """Pop value for a given metadata key."""
Original file line number Diff line number Diff line change @@ -63,6 +63,12 @@ def test_metadata_behave_as_dict(self):
6363 self .assertTrue ("foo" in metadata )
6464 self .assertFalse ("FOO" in metadata )
6565
66+ # default value
67+ # Should return default when key is not in metadata
68+ self .assertEqual (metadata .get ("FOO" , "default" ), "default" )
69+ # Should return default when key is in metadata but value is None
70+ metadata .FOO = None
71+ self .assertEqual (metadata .get ("FOO" , "default" ), "default" )
6672 # export
6773 self .assertEqual (str (metadata ), "foo: BAR" )
6874
You can’t perform that action at this time.
0 commit comments