Skip to content

Commit 654e8af

Browse files
committed
RF: Style updates
1 parent ee07713 commit 654e8af

File tree

2 files changed

+5
-13
lines changed

2 files changed

+5
-13
lines changed

nibabel/tests/test_wrapstruct.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -183,15 +183,10 @@ def test_mappingness(self):
183183
assert_equal(hdr.get(keys[0]), vals[0])
184184
assert_equal(hdr.get(keys[0], 'default'), vals[0])
185185

186-
# make sure .get returns values
187-
# which evaluate to False. We
188-
# have to use a different falsy
189-
# value depending on the data
190-
# type of the first header field.
191-
if np.issubdtype(hdr_dt[0], np.number):
192-
falsyval = 0
193-
else:
194-
falsyval = b''
186+
# make sure .get returns values which evaluate to False. We have to
187+
# use a different falsy value depending on the data type of the first
188+
# header field.
189+
falsyval = 0 if np.issubdtype(hdr_dt[0], np.number) else b''
195190

196191
hdr[keys[0]] = falsyval
197192
assert_equal(hdr[keys[0]], falsyval)

nibabel/wrapstruct.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -343,10 +343,7 @@ def items(self):
343343

344344
def get(self, k, d=None):
345345
''' Return value for the key k if present or d otherwise'''
346-
if k in self.keys():
347-
return self._structarr[k]
348-
else:
349-
return d
346+
return self._structarr[k] if k in self.keys() else d
350347

351348
def check_fix(self, logger=None, error_level=None):
352349
''' Check structured data with checks

0 commit comments

Comments
 (0)