Cleanup some old python 2 idioms

yt/config.py

@@ -109,7 +109,7 @@
     except IOError:
         warnings.warn("unable to write new config file")
 
-class YTConfigParser(configparser.ConfigParser, object):
+class YTConfigParser(configparser.ConfigParser):
     def __setitem__(self, key, val):
         self.set(key[0], key[1], val)
 

yt/data_objects/construction_data_containers.py

@@ -1073,7 +1073,7 @@ def _get_grid_bounds_size(self):
 
         return bounds, size
 
-class LevelState(object):
+class LevelState:
     current_dx = None
     current_dims = None
     current_level = None

yt/data_objects/derived_quantities.py

@@ -72,7 +72,7 @@ def process_chunk(self, data, *args, **kwargs):
     def reduce_intermediate(self, values):
         raise NotImplementedError
 
-class DerivedQuantityCollection(object):
+class DerivedQuantityCollection:
     def __new__(cls, data_source, *args, **kwargs):
         inst = object.__new__(cls)
         inst.data_source = data_source

yt/data_objects/level_sets/clump_info_items.py

@@ -8,7 +8,7 @@
 def add_clump_info(name, function):
     clump_info_registry[name] = ClumpInfoCallback(name, function)
 
-class ClumpInfoCallback(object):
+class ClumpInfoCallback:
     r"""
     A ClumpInfoCallback is a function that takes a clump, computes a 
     quantity, and returns a string to be printed out for writing clump info.

yt/data_objects/level_sets/clump_validators.py

@@ -12,7 +12,7 @@
 def add_validator(name, function):
     clump_validator_registry[name] = ClumpValidator(function)
 
-class ClumpValidator(object):
+class ClumpValidator:
     r"""
     A ClumpValidator is a function that takes a clump and returns 
     True or False as to whether the clump is valid and shall be kept.

yt/data_objects/octree_subset.py

@@ -446,7 +446,7 @@ def select_particles(self, selector, x, y, z):
         mask = selector.select_points(x,y,z, 0.0)
         return mask
 
-class OctreeSubsetBlockSlicePosition(object):
+class OctreeSubsetBlockSlicePosition:
     def __init__(self, ind, block_slice):
         self.ind = ind
         self.block_slice = block_slice
@@ -497,7 +497,7 @@ def _field_parameter_state(self, field_parameters):
         yield self.block_slice.octree_subset._field_parameter_state(
                 field_parameters)
 
-class OctreeSubsetBlockSlice(object):
+class OctreeSubsetBlockSlice:
     def __init__(self, octree_subset):
         self.octree_subset = octree_subset
         # Cache some attributes

yt/data_objects/particle_filters.py

@@ -9,13 +9,13 @@
 # One to one mapping
 filter_registry = {}
 
-class DummyFieldInfo(object):
+class DummyFieldInfo:
     particle_type = True
     sampling_type = 'particle'
 
 dfi = DummyFieldInfo()
 
-class ParticleFilter(object):
+class ParticleFilter:
     def __init__(self, name, function, requires, filtered_type):
         self.name = name
         self.function = function

yt/data_objects/particle_trajectories.py

@@ -11,7 +11,7 @@
 import numpy as np
 from yt.utilities.on_demand_imports import _h5py as h5py
 
-class ParticleTrajectories(object):
+class ParticleTrajectories:
     r"""A collection of particle trajectories in time over a series of
     datasets.
 

yt/data_objects/profiles.py

@@ -61,7 +61,7 @@ def save_state(*args, **kwargs):
         return tr
     return save_state
 
-class ProfileFieldAccumulator(object):
+class ProfileFieldAccumulator:
     def __init__(self, n_fields, size):
         shape = size + (n_fields,)
         self.values = np.zeros(shape, dtype="float64")

yt/data_objects/region_expression.py

@@ -5,7 +5,7 @@
 from yt.utilities.exceptions import YTDimensionalityError
 from yt.visualization.line_plot import LineBuffer
 
-class RegionExpression(object):
+class RegionExpression:
     _all_data = None
     def __init__(self, ds):
         self.ds = weakref.proxy(ds)

yt/data_objects/static_output.py

@@ -89,7 +89,7 @@ def __init__(cls, name, b, d):
         output_type_registry[name] = cls
         mylog.debug("Registering: %s as %s", name, cls)
 
-class IndexProxy(object):
+class IndexProxy:
     # This is a simple proxy for Index objects.  It enables backwards
     # compatibility so that operations like .h.sphere, .h.print_stats and
     # .h.grid_left_edge will correctly pass through to the various dataset or
@@ -107,7 +107,7 @@ def __getattr__(self, name):
             return getattr(self.ds.index, name)
         raise AttributeError
 
-class MutableAttribute(object):
+class MutableAttribute:
     """A descriptor for mutable data"""
     def __init__(self, display_array = False):
         self.data = weakref.WeakKeyDictionary()
@@ -1522,7 +1522,7 @@ def _reconstruct_ds(*args, **kwargs):
     return ds
 
 @functools.total_ordering
-class ParticleFile(object):
+class ParticleFile:
     def __init__(self, ds, io, filename, file_id, range = None):
         self.ds = ds
         self.io = weakref.proxy(io)

yt/data_objects/tests/test_derived_quantities.py

@@ -1,4 +1,4 @@
-from __future__ import division
+
 
 import numpy as np
 import yt

yt/data_objects/tests/test_octree.py

@@ -12,7 +12,7 @@ def test_building_tree():
     '''
     ds = fake_sph_grid_ds()
     octree = ds.octree(n_ref=1)
-    assert(type(octree) == YTOctree)
+    assert(type(octree) is YTOctree)
     assert(octree[('index', 'x')].shape[0] == 456)
 
 def test_saving_loading():

yt/data_objects/tests/test_particle_filter.py

@@ -1,4 +1,4 @@
-from __future__ import print_function
+
 
 import numpy as np
 import os

yt/data_objects/time_series.py

@@ -31,7 +31,7 @@
 from yt.utilities.parameter_file_storage import \
     simulation_time_series_registry
 
-class AnalysisTaskProxy(object):
+class AnalysisTaskProxy:
     def __init__(self, time_series):
         self.time_series = time_series
 
@@ -75,7 +75,7 @@ def get_filenames_from_glob_pattern(filenames):
          "omega_matter", "omega_lambda", "omega_radiation",
          "hubble_constant")
 
-class TimeSeriesParametersContainer(object):
+class TimeSeriesParametersContainer:
     def __init__(self, data_object):
         self.data_object = data_object
 
@@ -84,7 +84,7 @@ def __getattr__(self, attr):
             return self.data_object.eval(get_ds_prop(attr)())
         raise AttributeError(attr)
 
-class DatasetSeries(object):
+class DatasetSeries:
     r"""The DatasetSeries object is a container of multiple datasets,
     allowing easy iteration and computation on them.
 
@@ -449,7 +449,7 @@ def particle_trajectories(self, indices, fields=None, suppress_logging=False, pt
         return ParticleTrajectories(self, indices, fields=fields, suppress_logging=suppress_logging,
                                     ptype=ptype)
 
-class TimeSeriesQuantitiesContainer(object):
+class TimeSeriesQuantitiesContainer:
     def __init__(self, data_object, quantities):
         self.data_object = data_object
         self.quantities = quantities
@@ -465,7 +465,7 @@ def run_quantity(*args, **kwargs):
             return run_quantity
         return run_quantity_wrapper(q, key)
 
-class DatasetSeriesObject(object):
+class DatasetSeriesObject:
     def __init__(self, time_series, data_object_name, *args, **kwargs):
         self.time_series = weakref.proxy(time_series)
         self.data_object_name = data_object_name

yt/data_objects/unions.py

@@ -1,6 +1,6 @@
 from yt.funcs import ensure_list
 
-class Union(object):
+class Union:
     _union_type = ""
     def __init__(self, name, sub_types):
         self.name = name

yt/exthook.py

@@ -24,7 +24,7 @@
 import os
 
 
-class ExtensionImporter(object):
+class ExtensionImporter:
     """This importer redirects imports from this submodule to other locations.
     This makes it possible to transition from the old flaskext.name to the
     newer flask_name without people having a hard time.

yt/fields/derived_field.py

@@ -32,7 +32,7 @@ def _TranslationFunc(field, data):
 def NullFunc(field, data):
     raise YTFieldNotFound(field.name)
 
-class DerivedField(object):
+class DerivedField:
     """
     This is the base class used to describe a cell-by-cell derived field.
 
@@ -412,7 +412,7 @@ def get_latex_display_name(self):
         return label
 
 
-class FieldValidator(object):
+class FieldValidator:
     pass
 
 class ValidateParameter(FieldValidator):

yt/fields/domain_context.py

@@ -1,6 +1,6 @@
 domain_context_registry = {}
 
-class DomainContext(object):
+class DomainContext:
     class __metaclass__(type):
         def __init__(cls, name, b, d):
             type.__init__(cls, name, b, d)

yt/fields/field_detector.py

@@ -58,8 +58,8 @@ class fake_dataset(defaultdict):
             ds.periodicity = (True, True, True)
         self.ds = ds
 
-        class fake_index(object):
-            class fake_io(object):
+        class fake_index:
+            class fake_io:
                 def _read_data_set(io_self, data, field):
                     return self._read_data(field)
                 _read_exception = RuntimeError

yt/fields/field_info_container.py

@@ -401,7 +401,7 @@ def check_derived_fields(self, fields_to_check = None):
             except Exception as e:
                 if field in self._show_field_errors:
                     raise
-                if type(e) != YTFieldNotFound:
+                if not isinstance(e, YTFieldNotFound):
                     # if we're doing field tests, raise an error
                     # see yt.fields.tests.test_fields
                     if hasattr(self.ds, '_field_test_dataset'):

yt/fields/tests/test_fields.py

@@ -61,7 +61,7 @@ def _strip_ftype(field):
     return field[1]
 
 
-class TestFieldAccess(object):
+class TestFieldAccess:
     description = None
 
     def __init__(self, field_name, ds, nprocs):
@@ -405,19 +405,19 @@ def test_ion_field_labels():
     ds = fake_random_ds(16, fields=fields, units=units)
 
     # by default labels should use roman numerals
-    default_labels = {"O_p1_number_density":u"$\\rm{O\ II\ Number\ Density}$",
-                      "O2_p1_number_density":u"$\\rm{O_{2}\ II\ Number\ Density}$",
-                      "CO2_p1_number_density":u"$\\rm{CO_{2}\ II\ Number\ Density}$",
-                      "Co_p1_number_density":u"$\\rm{Co\ II\ Number\ Density}$",
-                      "O2_p2_number_density":u"$\\rm{O_{2}\ III\ Number\ Density}$",
-                      "H2O_p1_number_density":u"$\\rm{H_{2}O\ II\ Number\ Density}$"}
-
-    pm_labels = {"O_p1_number_density":u"$\\rm{{O}^{+}\ Number\ Density}$",
-                 "O2_p1_number_density":u"$\\rm{{O_{2}}^{+}\ Number\ Density}$",
-                 "CO2_p1_number_density":u"$\\rm{{CO_{2}}^{+}\ Number\ Density}$",
-                 "Co_p1_number_density":u"$\\rm{{Co}^{+}\ Number\ Density}$",
-                 "O2_p2_number_density":u"$\\rm{{O_{2}}^{++}\ Number\ Density}$",
-                 "H2O_p1_number_density":u"$\\rm{{H_{2}O}^{+}\ Number\ Density}$"}
+    default_labels = {"O_p1_number_density":"$\\rm{O\ II\ Number\ Density}$",
+                      "O2_p1_number_density":"$\\rm{O_{2}\ II\ Number\ Density}$",
+                      "CO2_p1_number_density":"$\\rm{CO_{2}\ II\ Number\ Density}$",
+                      "Co_p1_number_density":"$\\rm{Co\ II\ Number\ Density}$",
+                      "O2_p2_number_density":"$\\rm{O_{2}\ III\ Number\ Density}$",
+                      "H2O_p1_number_density":"$\\rm{H_{2}O\ II\ Number\ Density}$"}
+
+    pm_labels = {"O_p1_number_density":"$\\rm{{O}^{+}\ Number\ Density}$",
+                 "O2_p1_number_density":"$\\rm{{O_{2}}^{+}\ Number\ Density}$",
+                 "CO2_p1_number_density":"$\\rm{{CO_{2}}^{+}\ Number\ Density}$",
+                 "Co_p1_number_density":"$\\rm{{Co}^{+}\ Number\ Density}$",
+                 "O2_p2_number_density":"$\\rm{{O_{2}}^{++}\ Number\ Density}$",
+                 "H2O_p1_number_density":"$\\rm{{H_{2}O}^{+}\ Number\ Density}$"}
 
     fobj = ds.fields.stream
 

yt/fields/xray_emission_fields.py

@@ -55,7 +55,7 @@ def __str__(self):
         return self.msg
 
 
-class XrayEmissivityIntegrator(object):
+class XrayEmissivityIntegrator:
     r"""Class for making X-ray emissivity fields. Uses hdf5 data tables
     generated from Cloudy and AtomDB/APEC.
 

yt/frontends/art/data_structures.py

@@ -719,7 +719,7 @@ def fill(self, content, ftfields, selector):
                 source)
         return tr
 
-class ARTDomainFile(object):
+class ARTDomainFile:
     """
     Read in the AMR, left/right edges, fill out the octhandler
     """

yt/frontends/art/io.py

@@ -308,7 +308,7 @@ def _read_art_level_info(f, level_oct_offsets, level, coarse_grid=128,
     fl = np.ones((nLevel, 6), dtype='int64')
     iocts = np.zeros(nLevel+1, dtype='int64')
     idxa, idxb = 0, 0
-    chunk = long(1e6)  # this is ~111MB for 15 dimensional 64 bit arrays
+    chunk = int(1e6)  # this is ~111MB for 15 dimensional 64 bit arrays
     left = nLevel
     while left > 0:
         this_chunk = min(chunk, left)
@@ -472,7 +472,7 @@ def _read_child_mask_level(f, level_child_offsets, level, nLevel, nhydro_vars):
     ioctch = np.zeros(nLevel, dtype='uint8')
     idc = np.zeros(nLevel, dtype='int32')
 
-    chunk = long(1e6)
+    chunk = int(1e6)
     left = nLevel
     width = nhydro_vars+6
     a, b = 0, 0

yt/frontends/artio/definitions.py

@@ -30,7 +30,7 @@
 art_to_yt = dict(zip(yt_to_art.values(), yt_to_art.keys()))
 
 
-class ARTIOconstants():
+class ARTIOconstants:
     def __init__(self):
         self.yr = 365.25*86400
         self.Myr = 1.0e6*self.yr

yt/frontends/boxlib/data_structures.py

@@ -123,7 +123,7 @@ def _fill_child_mask(self, child, mask, tofill, dlevel=1):
              startIndex[2]:endIndex[2]] = tofill
 
 
-class BoxLibParticleHeader(object):
+class BoxLibParticleHeader:
 
     def __init__(self, ds, directory_name, is_checkpoint, 
                  extra_field_names=None):
@@ -220,7 +220,7 @@ def _generate_particle_fields(self, extra_field_names):
                                             for t in self.known_real_fields])
 
 
-class AMReXParticleHeader(object):
+class AMReXParticleHeader:
 
     def __init__(self, ds, directory_name, is_checkpoint, 
                  extra_field_names=None):
@@ -1405,7 +1405,7 @@ def _read_header(raw_file, field):
     return nghost, all_boxes, all_file_names, all_offsets
 
 
-class WarpXHeader(object):
+class WarpXHeader:
     def __init__(self, header_fn):
         self.data = {}
         with open(header_fn, "r") as f:

yt/frontends/chombo/io.py

@@ -33,7 +33,7 @@ def box_size(corners):
         self._offsets = {}
         num_comp = self._handle.attrs['num_components']
         level = 0
-        while 1:
+        while True:
             lname = 'level_%i' % level
             if lname not in self._handle: break
             boxes = self._handle['level_0']['boxes'][()]