qa

scripts/speed-test.py

@@ -107,7 +107,7 @@ def main():
     if path.endswith(".yaml"):
         with open(path, "r") as f:
             path = yaml.safe_load(f)
-        
+
     ds = open_dataset(path)
 
     total = len(ds)
@@ -124,9 +124,7 @@ def main():
         indexes = indexes[: args.count]
 
     if nworkers > 1:
-        tests = Tests(
-            [SpeedTest(path, indexes[i::nworkers]) for i in range(nworkers)]
-        )
+        tests = Tests([SpeedTest(path, indexes[i::nworkers]) for i in range(nworkers)])
 
         with Pool(nworkers) as pool:
             pool.map(tests, range(nworkers))

tests/test_create.py

@@ -42,9 +42,10 @@ def compare_zarr(dir1, dir2):
         assert a_ == b_, (a, b)
     for k, date in zip(range(a.shape[0]), a.dates):
         for j in range(a.shape[1]):
-            assert (a[k, j] == b[k, j]).all(), (k, date,j, a[k, j], b[k, j])
+            assert (a[k, j] == b[k, j]).all(), (k, date, j, a[k, j], b[k, j])
     compare(dir1, dir2)
 
+
 def compare(dir1, dir2):
     """Compare two directories recursively."""
     files1 = set(os.listdir(dir1))