Fix formatting of test_dataarray.TestDataArray.test_expand_dims_with_greater_dim_size and test_dataset.TestDataset.test_expand_dims to be pep8

Author: pletchm

xarray/tests/test_dataarray.py

@@ -1412,8 +1412,9 @@ def test_expand_dims_with_greater_dim_size(self):
                                          dim_1=['a', 'b', 'c'],
                                          x=np.linspace(0, 1, 3),
                                          dim_0=range(4)),
-                             dims=['y', 'z', 'dim_1', 'x', 'dim_0']
-                             ).drop(['y', 'dim_0']).assign_attrs(key="entry")
+                             dims=['y', 'z', 'dim_1', 'x', 'dim_0'],
+                             attrs={'key': 'entry'}
+                             ).drop(['y', 'dim_0'])
         assert_identical(expected, actual)
 
         # Test with kwargs instead of passing dict to dim arg.

xarray/tests/test_dataset.py

@@ -2064,21 +2064,20 @@ def test_expand_dims(self):
         # that DOES have coordinates.
         actual = original.expand_dims({"d": 4, "e": ["l", "m", "n"]})
 
-        expected = Dataset({
-            'x': xr.DataArray(original['x'].values * np.ones([4, 3, 3]),
-                              coords=dict(d=range(4),
-                                          e=['l', 'm', 'n'],
-                                          a=np.linspace(0, 1, 3)),
-                              dims=['d', 'e', 'a']
-                              ).drop('d'),
-            'y': xr.DataArray(original['y'].values * np.ones([4, 3, 4, 3]),
-                              coords=dict(d=range(4),
-                                          e=['l', 'm', 'n'],
-                                          b=np.linspace(0, 1, 4),
-                                          a=np.linspace(0, 1, 3)),
-                              dims=['d', 'e', 'b', 'a']
-                              ).drop('d')
-            }).assign_coords(c=np.linspace(0, 1, 5)).assign_attrs(key="entry")
+        expected = Dataset(
+            {'x': xr.DataArray(original['x'].values * np.ones([4, 3, 3]),
+                               coords=dict(d=range(4),
+                                           e=['l', 'm', 'n'],
+                                           a=np.linspace(0, 1, 3)),
+                               dims=['d', 'e', 'a']).drop('d'),
+             'y': xr.DataArray(original['y'].values * np.ones([4, 3, 4, 3]),
+                               coords=dict(d=range(4),
+                                           e=['l', 'm', 'n'],
+                                           b=np.linspace(0, 1, 4),
+                                           a=np.linspace(0, 1, 3)),
+                               dims=['d', 'e', 'b', 'a']).drop('d')},
+            coords={'c': np.linspace(0, 1, 5)},
+            attrs={'key': 'entry'})
         assert_identical(actual, expected)
 
         # Test with kwargs instead of passing dict to dim arg.