Black reformatting (#34)

* Reformat with black to handle line length issues

* Add line length check to config

Author: jleonatti

.ci/display_test.py

@@ -2,6 +2,6 @@
 import pyvista
 from pyvista.plotting import system_supports_plotting
 
-print('system_supports_plotting', system_supports_plotting())
+print("system_supports_plotting", system_supports_plotting())
 pyvista.OFF_SCREEN = True
 pyvista.plot(pyvista.Sphere())

.flake8

@@ -1,6 +1,5 @@
 [flake8]
-# To be added once code is refactored: E501
-select = W191, W291, W293, W391, E115, E117, E122, E124, E125, E225, E231, E301, E303, F401, F403
+select = W191, W291, W293, W391, E115, E117, E122, E124, E125, E225, E231, E301, E303, E501, F401, F403
 per-file-ignores = __init__.py:F401
 count = True
 max-complexity = 10

ansys/dpf/post/__init__.py

@@ -16,4 +16,4 @@
 
 """
 
-#dpf.core.start_local_server()
+# dpf.core.start_local_server()

ansys/dpf/post/_version.py

@@ -3,4 +3,4 @@
 version_info = 0, 2, 0
 
 # Nice string for the version
-__version__ = '.'.join(map(str, version_info))
+__version__ = ".".join(map(str, version_info))

ansys/dpf/post/common.py

@@ -19,7 +19,7 @@
 #     shell_bottom = 6
 
 
-class Grouping():
+class Grouping:
     """Class with Enum inheritance. Must be used to
     describe a grouping command when the API allows it.
 
@@ -29,18 +29,20 @@ class Grouping():
     >>> solution = post.solution("file.rst")
     >>> disp = solution.elemental_stress(element_shape = post.grouping.by_el_shape)
     """
+
     by_el_shape = "elshape"
     by_material = "mat"
     by_body = "body"
 
 
-class _AvailableKeywords():
+class _AvailableKeywords:
     """Contains all the keywords that can be used inside of
     a method from a post.solution(file_path) object.
 
     In order to view the complete list of available keywords, use:
         post.print_available_keywords()
     """
+
     location = "location"
     node_scoping = "node_scoping"
     element_scoping = "element_scoping"
@@ -56,43 +58,49 @@ class _AvailableKeywords():
     def __str__(self):
         txt = ""
         for attr in dir(_AvailableKeywords):
-            if not attr.startswith("__") and not attr.startswith("_") and not callable(getattr(_AvailableKeywords, attr)):
+            if (
+                not attr.startswith("__")
+                and not attr.startswith("_")
+                and not callable(getattr(_AvailableKeywords, attr))
+            ):
                 txt += attr
                 txt += ": "
                 txt += self._description_mapping(attr)
                 txt += "\n"
         return txt
 
     def _description_mapping(self, attr_name):
-        if (attr_name == self.location):
+        if attr_name == self.location:
             return "str. Use post.locations.(...) as helper."
-        if (attr_name == self.node_scoping):
+        if attr_name == self.node_scoping:
             return "list, int or dpf.core.Scoping"
-        if (attr_name == self.element_scoping):
+        if attr_name == self.element_scoping:
             return "list, int or dpf.core.Scoping"
-        if (attr_name == self.time_scoping):
+        if attr_name == self.time_scoping:
             return "list, int or dpf.core.Scoping"
-        if (attr_name == self.named_selection):
+        if attr_name == self.named_selection:
             return "str. Name of named_selection."
-        if (attr_name == self.time):
+        if attr_name == self.time:
             return "float"
-        if (attr_name == self.set):
+        if attr_name == self.set:
             return "int"
-        if (attr_name == self.mapdl_grouping):
+        if attr_name == self.mapdl_grouping:
             return "int. Write 186 to get mapdl_elements solid_186."
-        if (attr_name == self.grouping):
+        if attr_name == self.grouping:
             return "str. Use post.grouping.(...) as helper."
 
 
-class _AnalysisType():
+class _AnalysisType:
     """Contains Python analysis type names. For developers usage."""
+
     static = "static"
     modal = "modal"
     harmonic = "harmonic"
     transient = "transient"
 
 
-class _PhysicsType():
+class _PhysicsType:
     """Contains Python physics type names. For developers usage."""
+
     mecanic = "mecanic"
     thermal = "thermal"

ansys/dpf/post/displacement.py

@@ -4,8 +4,10 @@
 from ansys.dpf.core import locations
 from ansys.dpf.post.errors import NodalLocationError
 
+
 class Displacement(Vector):
     """Defines the displacement object, that is a vector object."""
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self._operator_name = "U"
@@ -22,6 +24,7 @@ def __str__(self):
 
 class ComplexDisplacement(ComplexVector):
     """Defines the complex displacement object, that is a vector object."""
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self._operator_name = "U"

ansys/dpf/post/dpf_solution.py

@@ -31,6 +31,7 @@
 
 class DpfSolution:
     """Main class of post result API."""
+
     def __init__(self, data_sources, model):
         """Initialization of the solution using data_sources
         and dpf.core.Model object."""
@@ -67,8 +68,11 @@ def _check_nodal_location(self, **kwargs):
                 raise NodalLocationError()
 
     def __str__(self):
-        txt = '%s solution object.' % self._model.metadata.result_info.analysis_type.capitalize() +\
-              '\n\n\nData Sources\n------------------------------\n'
+        txt = (
+            "%s solution object."
+            % self._model.metadata.result_info.analysis_type.capitalize()
+            + "\n\n\nData Sources\n------------------------------\n"
+        )
         ds_str = self._data_sources.__str__()
         txt += ds_str
         txt += "\n\n"
@@ -77,12 +81,11 @@ def __str__(self):
 
 
 class DpfMecanicSolution(DpfSolution):
-
     def __init__(self, data_sources, model):
         super().__init__(data_sources, model)
         self.misc = MecanicMisc(model, data_sources)
 
-    #result classes
+    # result classes
     def stress(self, **kwargs):
         """Returns a stress object from which it is possible to get ResultData.
 
@@ -113,7 +116,9 @@ def elastic_strain(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> elastic_strain = solution.elastic_strain(node_scoping = [1, 43])
         """
-        return ElasticStrain(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ElasticStrain(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def plastic_strain(self, **kwargs):
         """Returns a plastic strain object from which it is possible to get ResultData.
@@ -129,7 +134,9 @@ def plastic_strain(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> plastic_strain = solution.plastic_strain(node_scoping = [1, 43])
         """
-        return PlasticStrain(data_sources=self._data_sources, model=self._model, **kwargs)
+        return PlasticStrain(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def displacement(self, **kwargs):
         """Returns a displacement object from which it is possible to get ResultData.
@@ -145,7 +152,9 @@ def displacement(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> displacement = solution.displacement(node_scoping = [1, 43])
         """
-        return Displacement(data_sources=self._data_sources, model=self._model, **kwargs)
+        return Displacement(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def structural_temperature(self, **kwargs):
         """Returns a temperature object from which it is possible to get ResultData.
@@ -161,11 +170,14 @@ def structural_temperature(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> temperature = solution.structural_temperature(node_scoping = [1, 43])
         """
-        return StructuralTemperature(data_sources=self._data_sources, model=self._model, **kwargs)
+        return StructuralTemperature(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
 
 class DpfMecanicComplexSolution(DpfSolution):
     """Main class of post solution if the analysis gives complex solution (Modal, Harmonic)."""
+
     def __init__(self, data_sources, model):
         super().__init__(data_sources, model)
         self.misc = ComplexMecanicMisc(model, data_sources)
@@ -185,7 +197,7 @@ def has_complex_result(self):
         tfq_sup = self._model.metadata.time_freq_support
         if not tfq_sup:
             return False
-        if (tfq_sup.complex_frequencies == None):
+        if tfq_sup.complex_frequencies == None:
             return False
         return True
 
@@ -203,7 +215,9 @@ def displacement(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> displacement = solution.displacement(node_scoping = [1, 43])
         """
-        return ComplexDisplacement(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ComplexDisplacement(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def structural_temperature(self, **kwargs):
         """Returns a temperature object from which it is possible to get ResultData.
@@ -219,7 +233,9 @@ def structural_temperature(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> temperature = solution.structural_temperature(node_scoping = [1, 43])
         """
-        return ComplexStructuralTemperature(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ComplexStructuralTemperature(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def plastic_strain(self, **kwargs):
         """Returns a plastic strain object from which it is possible to get ResultData.
@@ -235,7 +251,9 @@ def plastic_strain(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> plastic_strain = solution.plastic_strain(node_scoping = [1, 43])
         """
-        return ComplexPlasticStrain(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ComplexPlasticStrain(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def elastic_strain(self, **kwargs):
         """Returns an elastic strain object from which it is possible to get ResultData.
@@ -251,7 +269,9 @@ def elastic_strain(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> elastic_strain = solution.elastic_strain(node_scoping = [1, 43])
         """
-        return ComplexElasticStrain(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ComplexElasticStrain(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def stress(self, **kwargs):
         """Returns a stress object from which it is possible to get ResultData.
@@ -267,14 +287,17 @@ def stress(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> stress = solution.stress(node_scoping = [1, 43])
         """
-        return ComplexStress(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ComplexStress(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
 
 class DpfThermalSolution(DpfSolution):
     """Main class of post solution if thermal analysis."""
+
     def __init__(self, data_sources, model):
         super().__init__(data_sources, model)
-        #self.misc = ThermalMisc(model, data_sources)
+        # self.misc = ThermalMisc(model, data_sources)
 
     def __str__(self):
         txt = super().__str__()
@@ -330,7 +353,9 @@ def electric_potential(self, **kwargs):
         >>> ep = solution.electric_potential(node_scoping = [1, 43])
         """
         self._check_nodal_location(**kwargs)
-        return ElectricPotential(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ElectricPotential(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )
 
     def electric_field(self, **kwargs):
         """Returns an electric field object from which it is possible to get ResultData.
@@ -346,4 +371,6 @@ def electric_field(self, **kwargs):
         >>> solution = post.load_solution(file.rst)
         >>> ef = solution.electric_field(node_scoping = [1, 43])
         """
-        return ElectricField(data_sources=self._data_sources, model=self._model, **kwargs)
+        return ElectricField(
+            data_sources=self._data_sources, model=self._model, **kwargs
+        )

ansys/dpf/post/electric_results.py

@@ -7,13 +7,16 @@
 
 class ElectricField(Vector):
     """Defines the temperature object for thermal/electric analysis, that is a scalar object."""
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self._operator_name = "EF"
 
-        #disable element scoping
+        # disable element scoping
         if _AvailableKeywords.element_scoping in kwargs:
-            raise Exception("Element scoping is not available with thermal/electric results.")
+            raise Exception(
+                "Element scoping is not available with thermal/electric results."
+            )
         self.definition._Definition__element_scoping_locked = True
 
     def __str__(self):
@@ -25,13 +28,16 @@ def __str__(self):
 
 class ElectricPotential(Scalar):
     """Defines the temperature object for thermal/electric analysis, that is a scalar object."""
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self._operator_name = "VOLT"
 
-        #disable element scoping
+        # disable element scoping
         if _AvailableKeywords.element_scoping in kwargs:
-            raise Exception("Element scoping is not available with thermal/electric results.")
+            raise Exception(
+                "Element scoping is not available with thermal/electric results."
+            )
         self.definition._Definition__element_scoping_locked = True
 
     def __str__(self):

ansys/dpf/post/errors.py

@@ -1,7 +1,8 @@
 """DPF-Post specific errors"""
 
+
 class NodalLocationError(ValueError):
-    """Raised when attempting to """
+    """Raised when attempting to"""
 
     def __init__(self, msg="The location must be nodal."):
         ValueError.__init__(self, msg)