From e5aff065fb5b8d8e25eacc7173117851b697c5b4 Mon Sep 17 00:00:00 2001
From: Alex Merose <al@merose.com>
Date: Sun, 24 Mar 2024 15:55:58 +0530
Subject: [PATCH] Reformat.

---
 xarray_sql/df_test.py | 30 +++++++++++++++++-------------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/xarray_sql/df_test.py b/xarray_sql/df_test.py
index 9007145..59953e6 100644
--- a/xarray_sql/df_test.py
+++ b/xarray_sql/df_test.py
@@ -19,18 +19,18 @@ def rand_wx(start: str, end: str) -> xr.Dataset:
   temperature = 15 + 8 * np.random.randn(720, 1440, len(time), len(level))
   precipitation = 10 * np.random.rand(720, 1440, len(time), len(level))
   return xr.Dataset(
-    data_vars=dict(
-      temperature=(['lat', 'lon', 'time', 'level'], temperature),
-      precipitation=(['lat', 'lon', 'time', 'level'], precipitation),
-    ),
-    coords=dict(
-      lat=lat,
-      lon=lon,
-      time=time,
-      level=level,
-      reference_time=reference_time,
-    ),
-    attrs=dict(description='Random weather.')
+      data_vars=dict(
+          temperature=(['lat', 'lon', 'time', 'level'], temperature),
+          precipitation=(['lat', 'lon', 'time', 'level'], precipitation),
+      ),
+      coords=dict(
+          lat=lat,
+          lon=lon,
+          time=time,
+          level=level,
+          reference_time=reference_time,
+      ),
+      attrs=dict(description='Random weather.'),
   )
 
 
@@ -115,7 +115,11 @@ def test_column_metadata_preserved(self):
     try:
       _ = read_xarray(self.randwx, chunks=dict(time=24)).compute()
     except ValueError as e:
-      if 'The columns in the computed data do not match the columns in the provided metadata' in str(e):
+      if (
+          'The columns in the computed data do not match the columns in the'
+          ' provided metadata'
+          in str(e)
+      ):
         self.fail('Column metadata is incorrect.')