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):
221
221
222
222
def get (self , key : str , default : Any = None ) -> Any :
223
223
"""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
225
226
226
227
def pop (self , key : str , default : Any = None ) -> Any :
227
228
"""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):
63
63
self .assertTrue ("foo" in metadata )
64
64
self .assertFalse ("FOO" in metadata )
65
65
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" )
66
72
# export
67
73
self .assertEqual (str (metadata ), "foo: BAR" )
68
74
You can’t perform that action at this time.
0 commit comments