black pass

Author: convert-repo

benchmarks/benchmarks/benchmarks.py

@@ -3,6 +3,7 @@
 import numpy as np
 from yt import YTArray, YTQuantity
 
+
 def time_quantity_init_scalar1():
     3.0 * YTQuantity(1, "m/s")
 

benchmarks/benchmarks/data_objects.py

@@ -1,15 +1,17 @@
 import numpy as np
 import yt
 
-if yt.__version__.startswith('2'):
+if yt.__version__.startswith("2"):
     from yt.mods import load, ColorTransferFunction
 # else:
 #     from yt.visualization.volume_rendering.old_camera import Camera
 
+
 class Suite:
     dsname = "HiresIsolatedGalaxy/DD0044/DD0044"
+
     def setup(self):
-        if yt.__version__.startswith('3'):
+        if yt.__version__.startswith("3"):
             self.ds = yt.load(self.dsname)
             self.ad = self.ds.all_data()
             self.field_name = "density"
@@ -19,46 +21,56 @@ def setup(self):
             self.field_name = "Density"
         # Warmup hdd
         self.ad[self.field_name]
-        if yt.__version__.startswith('3'):
-            mi, ma = self.ad.quantities['Extrema'](self.field_name)
-            self.tf = yt.ColorTransferFunction((np.log10(mi)+1, np.log10(ma)))
+        if yt.__version__.startswith("3"):
+            mi, ma = self.ad.quantities["Extrema"](self.field_name)
+            self.tf = yt.ColorTransferFunction((np.log10(mi) + 1, np.log10(ma)))
         else:
-            mi, ma = self.ad.quantities['Extrema'](self.field_name)[0]
-            self.tf = ColorTransferFunction((np.log10(mi)+1, np.log10(ma)))
+            mi, ma = self.ad.quantities["Extrema"](self.field_name)[0]
+            self.tf = ColorTransferFunction((np.log10(mi) + 1, np.log10(ma)))
         self.tf.add_layers(5, w=0.02, colormap="spectral")
         self.c = [0.5, 0.5, 0.5]
         self.L = [0.5, 0.2, 0.7]
         self.W = 1.0
         self.Npixels = 512
 
-    if yt.__version__.startswith('3'):
+    if yt.__version__.startswith("3"):
+
         def time_load_all_data(self):
             dd = self.ds.all_data()
             dd[self.field_name]
+
     else:
+
         def time_load_all_data(self):
             self.ds.h.all_data()
             dd[self.field_name]
 
     def time_extrema_quantities(self):
-        self.ad.quantities['Extrema'](self.field_name)
+        self.ad.quantities["Extrema"](self.field_name)
+
+    if yt.__version__.startswith("3"):
 
-    if yt.__version__.startswith('3'):
         def time_alldata_projection(self):
             self.ds.proj(self.field_name, 0)
+
     else:
+
         def time_alldata_projection(self):
-           self.ds.h.proj(0, self.field_name)
+            self.ds.h.proj(0, self.field_name)
+
+    if yt.__version__.startswith("3"):
 
-    if yt.__version__.startswith('3'):
         def time_slice(self):
             slc = self.ds.slice(0, 0.5)
             slc[self.field_name]
+
     else:
+
         def time_slice(self):
             slc = self.ds.h.slice(0, 0.5, self.field_name)
             slc[self.field_name]
 
+
 #    if yt.__version__.startswith('3'):
 #        def command(self):
 #            cam = Camera(self.c, self.L, self.W, self.Npixels, self.tf, ds=self.ds)

benchmarks/benchmarks/large_tipsy.py

@@ -2,18 +2,24 @@
 import yt
 from yt.utilities.answer_testing.framework import data_dir_load
 
+
 class PKDGravTipsySuite:
     dsname = "halo1e11_run1.00400/halo1e11_run1.00400"
     timeout = 360.0
+
     def setup(self):
-        cosmology_parameters = dict(current_redshift = 0.0,
-                                    omega_lambda = 0.728,
-                                    omega_matter = 0.272,
-                                    hubble_constant = 0.702)
-        kwargs = dict(field_dtypes = {"Coordinates": "d"},
-                      cosmology_parameters = cosmology_parameters,
-                      unit_base = {'length': (1.0/60.0, "Mpccm/h")},
-                      n_ref = 64)
+        cosmology_parameters = dict(
+            current_redshift=0.0,
+            omega_lambda=0.728,
+            omega_matter=0.272,
+            hubble_constant=0.702,
+        )
+        kwargs = dict(
+            field_dtypes={"Coordinates": "d"},
+            cosmology_parameters=cosmology_parameters,
+            unit_base={"length": (1.0 / 60.0, "Mpccm/h")},
+            n_ref=64,
+        )
         self.ds = data_dir_load(self.dsname, yt.TipsyDataset, (), kwargs)
 
     def time_all_particles(self):
@@ -36,17 +42,23 @@ def time_particle_quantities(self):
         dd = self.ds.all_data()
         dd.quantities.extrema("particle_mass")
         dd.quantities.extrema("particle_velocity_magnitude")
-        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in 'xyz'])
+        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in "xyz"])
+
 
 class GasolineTipsySuite(PKDGravTipsySuite):
     dsname = "agora_1e11.00400/agora_1e11.00400"
     timeout = 360.0
+
     def setup(self):
-        cosmology_parameters = dict(current_redshift = 0.0,
-                                    omega_lambda = 0.728,
-                                    omega_matter = 0.272,
-                                    hubble_constant = 0.702)
-        kwargs = dict(cosmology_parameters = cosmology_parameters,
-                      unit_base = {'length': (1.0/60.0, "Mpccm/h")},
-                      n_ref = 64)
+        cosmology_parameters = dict(
+            current_redshift=0.0,
+            omega_lambda=0.728,
+            omega_matter=0.272,
+            hubble_constant=0.702,
+        )
+        kwargs = dict(
+            cosmology_parameters=cosmology_parameters,
+            unit_base={"length": (1.0 / 60.0, "Mpccm/h")},
+            n_ref=64,
+        )
         self.ds = data_dir_load(self.dsname, yt.TipsyDataset, (), kwargs)

benchmarks/benchmarks/small_enzo.py

@@ -1,8 +1,10 @@
 import numpy as np
 import yt
 
+
 class SmallEnzoSuite:
     dsname = "IsolatedGalaxy/galaxy0030/galaxy0030"
+
     def setup(self):
         self.ds = yt.load(self.dsname)
 
@@ -38,7 +40,7 @@ def time_particle_quantities(self):
         dd = self.ds.all_data()
         dd.quantities.extrema("particle_mass")
         dd.quantities.extrema("particle_velocity_magnitude")
-        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in 'xyz'])
+        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in "xyz"])
 
     def time_gas_quantities(self):
         dd = self.ds.all_data()

benchmarks/benchmarks/small_flash.py

@@ -1,8 +1,10 @@
 import numpy as np
 import yt
 
+
 class SmallFlashSuite:
     dsname = "GasSloshingLowRes/sloshing_low_res_hdf5_plt_cnt_0690"
+
     def setup(self):
         self.ds = yt.load(self.dsname)
 
@@ -21,8 +23,7 @@ def time_project_weight(self):
         proj = self.ds.proj("density", 0, "density")
 
     def time_ghostzones(self):
-        dd = self.ds.sphere(self.ds.domain_center,
-                            self.ds.domain_width[0] * 0.25)
+        dd = self.ds.sphere(self.ds.domain_center, self.ds.domain_width[0] * 0.25)
         dd["velocity_divergence"]
 
     def time_gas_quantities(self):

benchmarks/benchmarks/small_gadget.py

@@ -1,6 +1,7 @@
 import numpy as np
 import yt
 
+
 class SmallGadgetSuite:
     dsname = "snapshot_033/snap_033.0.hdf5"
     timeout = 360.0
@@ -32,4 +33,4 @@ def time_particle_quantities(self):
         dd = self.ds.all_data()
         dd.quantities.extrema("particle_mass")
         dd.quantities.extrema("particle_velocity_magnitude")
-        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in 'xyz'])
+        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in "xyz"])

benchmarks/benchmarks/small_ramses.py

@@ -1,8 +1,10 @@
 import numpy as np
 import yt
 
+
 class SmallRAMSESSuite:
     dsname = "output_00080/info_00080.txt"
+
     def setup(self):
         self.ds = yt.load(self.dsname)
 
@@ -34,7 +36,7 @@ def time_particle_quantities(self):
         dd = self.ds.all_data()
         dd.quantities.extrema("particle_mass")
         dd.quantities.extrema("particle_velocity_magnitude")
-        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in 'xyz'])
+        dd.quantities.extrema(["particle_velocity_%s" % ax for ax in "xyz"])
 
     def time_gas_quantities(self):
         dd = self.ds.all_data()

conftest.py

@@ -22,32 +22,25 @@
 answer_files = {}
 
 # List of answer files
-answer_file_list = 'tests/tests.yaml'
-answer_dir = os.path.join(ytcfg.get('yt', 'test_data_dir'), 'answers')
+answer_file_list = "tests/tests.yaml"
+answer_dir = os.path.join(ytcfg.get("yt", "test_data_dir"), "answers")
 
 
 def pytest_addoption(parser):
     """
     Lets options be passed to test functions.
     """
     parser.addoption(
-        "--with-answer-testing",
-        action="store_true",
-        default=False,
+        "--with-answer-testing", action="store_true", default=False,
     )
     parser.addoption(
-        "--answer-store",
-        action="store_true",
-        default=False,
+        "--answer-store", action="store_true", default=False,
     )
     parser.addoption(
-        "--answer-big-data",
-        action="store_true",
-        default=False,
+        "--answer-big-data", action="store_true", default=False,
     )
     parser.addoption(
-        "--save-answer-arrays",
-        action="store_true",
+        "--save-answer-arrays", action="store_true",
     )
 
 
@@ -62,12 +55,13 @@ def pytest_configure(config):
         os.mkdir(answer_dir)
     # Read the list of answer test classes and their associated answer
     # file
-    with open(answer_file_list, 'r') as f:
+    with open(answer_file_list, "r") as f:
         answer_files = yaml.safe_load(f)
     # Register custom marks for answer tests and big data
-    config.addinivalue_line('markers', 'answer_test: Run the answer tests.')
-    config.addinivalue_line('markers', 'big_data: Run answer tests that require'
-        ' large data files.')
+    config.addinivalue_line("markers", "answer_test: Run the answer tests.")
+    config.addinivalue_line(
+        "markers", "big_data: Run answer tests that require" " large data files."
+    )
 
 
 def pytest_collection_modifyitems(config, items):
@@ -81,21 +75,23 @@ def pytest_collection_modifyitems(config, items):
     for item in items:
         # If it's an answer test and the appropriate CL option hasn't
         # been set, skip it
-        if "answer_test" in item.keywords and not config.getoption("--with-answer-testing"):
+        if "answer_test" in item.keywords and not config.getoption(
+            "--with-answer-testing"
+        ):
             item.add_marker(skip_answer)
         # If it's an answer test that requires big data and the CL
         # option hasn't been set, skip it
         if "big_data" in item.keywords and not config.getoption("--answer-big-data"):
             item.add_marker(skip_big)
 
 
-@pytest.fixture(scope='function')
+@pytest.fixture(scope="function")
 def temp_dir():
     r"""
     Creates a temporary directory needed by certain tests.
     """
     curdir = os.getcwd()
-    if int(os.environ.get('GENERATE_YTDATA', 0)):
+    if int(os.environ.get("GENERATE_YTDATA", 0)):
         tmpdir = os.getcwd()
     else:
         tmpdir = tempfile.mkdtemp()
@@ -106,7 +102,7 @@ def temp_dir():
         shutil.rmtree(tmpdir)
 
 
-@pytest.fixture(scope='class')
+@pytest.fixture(scope="class")
 def answer_file(request):
     r"""
     Assigns the name of the appropriate answer file as an attribute of
@@ -145,12 +141,11 @@ def answer_file(request):
         answer_file = answer_files[request.cls.__name__]
         # Make sure we're not overwriting an existing answer set
         if os.path.isfile(os.path.join(answer_dir, answer_file)):
-            if request.config.getoption('--answer-store'):
-                raise FileExistsError("Error, attempting to overwrite "
+            if request.config.getoption("--answer-store"):
+                raise FileExistsError(
+                    "Error, attempting to overwrite "
                     "answer file {}. Either specify a new version or "
-                    "set the `--force-override-answers` option".format(
-                        answer_file
-                    )
+                    "set the `--force-override-answers` option".format(answer_file)
                 )
     else:
         assert False
@@ -167,13 +162,13 @@ def _param_list(request):
     # co_varnames is all of the variable names local to the function
     # starting with self, then the passed args, then the vars defined
     # in the function body. This excludes fixture names
-    args = func.__code__.co_varnames[1:func.__code__.co_argcount]
+    args = func.__code__.co_varnames[1 : func.__code__.co_argcount]
     # funcargs includes the names and values of all arguments, including
     # fixtures, so we use args to weed out the fixtures. Need to have
     # special treatment of the data files loaded in fixtures for the
     # frontends
     for key, val in request.node.funcargs.items():
-        if key in args and not key.startswith('ds_'):
+        if key in args and not key.startswith("ds_"):
             test_params[key] = val
     # Convert python-specific data objects (such as tuples) to a more
     # io-friendly format (in order to not have python-specific anchors
@@ -182,7 +177,7 @@ def _param_list(request):
     return test_params
 
 
-@pytest.fixture(scope='function')
+@pytest.fixture(scope="function")
 def hashing(request):
     r"""
     Handles initialization, generation, and saving of answer test
@@ -244,12 +239,20 @@ def hashing(request):
     # Add the other test parameters
     hashes.update(params)
     # Add the function name as the "master" key to the hashes dict
-    hashes = {request.node.name : hashes}
+    hashes = {request.node.name: hashes}
     # Either save or compare
-    utils._handle_hashes(answer_dir, request.cls.answer_file, hashes,
-        request.config.getoption('--answer-store'))
-    if request.config.getoption('--save-answer-arrays'):
+    utils._handle_hashes(
+        answer_dir,
+        request.cls.answer_file,
+        hashes,
+        request.config.getoption("--answer-store"),
+    )
+    if request.config.getoption("--save-answer-arrays"):
         # answer_file has .yaml appended to it, but here we're saving
         # the arrays as .npy files, so we remove the .yaml extension
-        utils._save_arrays(answer_dir, request.cls.answer_file.split('.')[0],
-            request.cls.hashes, request.config.getoption('--answer-store'))
+        utils._save_arrays(
+            answer_dir,
+            request.cls.answer_file.split(".")[0],
+            request.cls.hashes,
+            request.config.getoption("--answer-store"),
+        )