Formatting

Author: Jenny Wong

eccodes/highlevel/message.py

@@ -68,10 +68,10 @@ def get(self, name, default=None, ktype=None):
             return default
 
     def set(self, *args):
-        """If two arguments are given, assumes this takes form of a single key 
-        value pair and sets the value of the given key. If a dictionary is passed 
-        then sets the values of all keys in the dictionary. Note, ordering 
-        if the keys is important. Finally, checks if value 
+        """If two arguments are given, assumes this takes form of a single key
+        value pair and sets the value of the given key. If a dictionary is passed
+        then sets the values of all keys in the dictionary. Note, ordering
+        if the keys is important. Finally, checks if value
         has been set correctly
 
         Raises
@@ -94,11 +94,11 @@ def set(self, *args):
         for name, value in key_values.items():
             with raise_keyerror(name):
                 eccodes.codes_set(self._handle, name, value)
-        
+
         # Check values just set
         for name, value in key_values.items():
             saved_value = self.get(name)
-            cast_value = value 
+            cast_value = value
             if not isinstance(value, type(saved_value)):
                 cast_value = type(saved_value)(value)
             if saved_value != cast_value:

tests/test_highlevel.py

@@ -64,22 +64,24 @@ def test_message_set():
         assert np.all(message.get("values") == vals)
         assert message.is_missing(missing_key)
 
+
 def test_message_set_multiple():
-     with eccodes.FileReader(TEST_GRIB_DATA) as reader:
+    with eccodes.FileReader(TEST_GRIB_DATA) as reader:
         message = next(reader)
         message.set({
             "centre": "ecmf",
-            "numberOfValues": 10, 
-            "shortName": "z", 
-            })
+            "numberOfValues": 10,
+            "shortName": "z",
+        })
         with pytest.raises(TypeError):
             message.set("centre", "ecmwf", 2)
         with pytest.raises(ValueError):
             message.set("stepRange", "0-12")
         message.set({
-            "stepType": "max", 
+            "stepType": "max",
             "stepRange": "0-12"
-            })
+        })
+
 
 def test_message_iter():
     with eccodes.FileReader(TEST_GRIB_DATA2) as reader: