blackened

Author: renefritze

.ci/make_env_file.py

@@ -3,13 +3,14 @@
 import os
 from os.path import expanduser
 from shlex import quote
+
 home = expanduser("~")
 
-prefixes = os.environ.get('ENV_PREFIXES', 'TRAVIS CI encrypt TOKEN TESTS').split(' ')
-blacklist = ['TRAVIS_COMMIT_MESSAGE']
-env_file = os.environ.get('ENV_FILE', os.path.join(home, 'env'))
-with open(env_file, 'wt') as env:
-    for k,v in os.environ.items():
+prefixes = os.environ.get("ENV_PREFIXES", "TRAVIS CI encrypt TOKEN TESTS").split(" ")
+blacklist = ["TRAVIS_COMMIT_MESSAGE"]
+env_file = os.environ.get("ENV_FILE", os.path.join(home, "env"))
+with open(env_file, "wt") as env:
+    for k, v in os.environ.items():
         for pref in prefixes:
             if k.startswith(pref) and k not in blacklist:
-                env.write('{}={}\n'.format(k,quote(v)))
+                env.write("{}={}\n".format(k, quote(v)))

lib/test_bindings.py

@@ -6,6 +6,7 @@
 import numpy as np
 import pydealii_bindings as dealii
 
+
 def test_vector():
     v = dealii.Vector(10)
     u = dealii.Vector(10)
@@ -38,9 +39,10 @@ def test_vector():
     ddones[:] = np.ones((100,), np.double)
     npdd = np.array(ddones, copy=False)
     assert np.allclose(npdd, np.ones((100,), dtype=np.double))
-    npdd += 1.
-    ddones /= 2.
+    npdd += 1.0
+    ddones /= 2.0
     assert np.allclose(npdd, ddones)
 
+
 if __name__ == "__main__":
     test_vector()

setup.cfg

@@ -68,5 +68,5 @@ relative_files = True
 source =
   src/pymor_dealii
   src/test
-omit = 
+omit =
   src/pymor_dealii/version.py

setup.py

@@ -3,6 +3,7 @@
 import os
 import pathlib
 import sys
+
 # versioneer (+dependencies) does not work in a pep518/7 context w/o modification here
 sys.path.append(os.path.dirname(__file__))
 import versioneer  # noqa
@@ -13,14 +14,12 @@
 
 
 class CMakeExtension(Extension):
-
     def __init__(self, name):
         # don't invoke the original build_ext for this special extension
         super().__init__(name, sources=[])
 
 
 class build_ext(build_ext_orig):
-
     def run(self):
         for ext in self.extensions:
             self.build_cmake(ext)
@@ -37,39 +36,37 @@ def build_cmake(self, ext):
         extdir.mkdir(parents=True, exist_ok=True)
 
         # example of cmake args
-        config = 'Debug' if self.debug else 'Release'
+        config = "Debug" if self.debug else "Release"
         cmake_args = [
-            '-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + str(extdir.parent.absolute()),
-            '-DCMAKE_BUILD_TYPE=' + config
+            "-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + str(extdir.parent.absolute()),
+            "-DCMAKE_BUILD_TYPE=" + config,
         ]
 
         # example of build args
-        build_args = [
-            '--config', config,
-            '--', '-j4'
-        ]
+        build_args = ["--config", config, "--", "-j4"]
 
         os.chdir(str(build_temp))
-        self.spawn(['cmake', str(cwd)] + cmake_args)
+        self.spawn(["cmake", str(cwd)] + cmake_args)
         if not self.dry_run:
-            self.spawn(['cmake', '--build', '.'] + build_args)
+            self.spawn(["cmake", "--build", "."] + build_args)
         os.chdir(str(cwd))
 
-cmdclass=versioneer.get_cmdclass()
-cmdclass['build_ext'] = build_ext
+
+cmdclass = versioneer.get_cmdclass()
+cmdclass["build_ext"] = build_ext
 
 setup(
     version=versioneer.get_version(),
-    name='pymor_dealii',
-    author='pyMOR developers',
-    author_email='[email protected]',
-    python_requires='>=3.7',
-    maintainer='Rene Fritze',
-    maintainer_email='[email protected]',
+    name="pymor_dealii",
+    author="pyMOR developers",
+    author_email="[email protected]",
+    python_requires=">=3.7",
+    maintainer="Rene Fritze",
+    maintainer_email="[email protected]",
     install_requires=[],
-    package_dir={'': 'src'},
-    packages=find_packages('src'),
+    package_dir={"": "src"},
+    packages=find_packages("src"),
     include_package_data=True,
-    ext_modules=[CMakeExtension('pymor_dealii')],
+    ext_modules=[CMakeExtension("pymor_dealii")],
     cmdclass=cmdclass,
 )

src/pymor_dealii/__init__.py

@@ -1,4 +1,4 @@
-
 from .version import get_versions
-__version__ = get_versions()['version']
+
+__version__ = get_versions()["version"]
 del get_versions

src/pymor_dealii/pymor/demo.py

@@ -6,6 +6,7 @@
 
 # instantiate deal.II model
 from dealii_elasticity import ElasticityExample
+
 cpp_disc = ElasticityExample(refine_steps=7)
 
 
@@ -17,55 +18,66 @@
 
 def run(plot_error=True):
     d = StationaryModel(
-        operator=LincombOperator([DealIIMatrixOperator(cpp_disc.lambda_mat()), DealIIMatrixOperator(cpp_disc.mu_mat())],
-                                [ProjectionParameterFunctional('lambda'), ProjectionParameterFunctional('mu')]),
-
+        operator=LincombOperator(
+            [
+                DealIIMatrixOperator(cpp_disc.lambda_mat()),
+                DealIIMatrixOperator(cpp_disc.mu_mat()),
+            ],
+            [
+                ProjectionParameterFunctional("lambda"),
+                ProjectionParameterFunctional("mu"),
+            ],
+        ),
         rhs=VectorOperator(DealIIVectorSpace.make_array([cpp_disc.rhs()])),
-
-        products={'energy': DealIIMatrixOperator(cpp_disc.mu_mat())},
-
-        visualizer=DealIIVisualizer(cpp_disc)
+        products={"energy": DealIIMatrixOperator(cpp_disc.mu_mat())},
+        visualizer=DealIIVisualizer(cpp_disc),
     )
-    parameter_space = d.parameters.space((1,10))
-
+    parameter_space = d.parameters.space((1, 10))
 
     # choose reduction method
     reductor = CoerciveRBReductor(
         d,
         product=d.energy_product,
-        coercivity_estimator=ExpressionParameterFunctional("max(mu)", d.parameters)
+        coercivity_estimator=ExpressionParameterFunctional("max(mu)", d.parameters),
     )
 
-
     # greedy basis generation
-    greedy_data = rb_greedy(d, reductor, parameter_space.sample_uniformly(3),
-                        extension_params={'method': 'gram_schmidt'}, max_extensions=5)
-
+    greedy_data = rb_greedy(
+        d,
+        reductor,
+        parameter_space.sample_uniformly(3),
+        extension_params={"method": "gram_schmidt"},
+        max_extensions=5,
+    )
 
     # get reduced order model
-    rd = greedy_data['rom']
-
+    rd = greedy_data["rom"]
 
     # validate reduced order model
-    result = reduction_error_analysis(rd, d, reductor,
-                                    test_mus=parameter_space.sample_randomly(10),
-                                    basis_sizes=reductor.bases['RB'].dim + 1,
-                                    condition=True, error_norms=[d.energy_norm],
-                                    plot=plot_error)
-
+    result = reduction_error_analysis(
+        rd,
+        d,
+        reductor,
+        test_mus=parameter_space.sample_randomly(10),
+        basis_sizes=reductor.bases["RB"].dim + 1,
+        condition=True,
+        error_norms=[d.energy_norm],
+        plot=plot_error,
+    )
 
     # visualize solution for parameter with maximum reduction error
-    mu_max = result['max_error_mus'][0, -1]
+    mu_max = result["max_error_mus"][0, -1]
     U = d.solve(mu_max)
     U_rb = reductor.reconstruct(rd.solve(mu_max))
     return result, U, U_rb, d
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     # print/plot results of validation
     from matplotlib import pyplot as plt
+
     result, U, U_rb, d = run()
-    print(result['summary'])
+    print(result["summary"])
     ERR = U - U_rb
-    d.visualize([U, U_rb, ERR], legend=['fom', 'rom', 'error'])
+    d.visualize([U, U_rb, ERR], legend=["fom", "rom", "error"])
     plt.show()

src/pymor_dealii/pymor/gui.py

@@ -7,11 +7,18 @@
 
 
 class DealIIVisualizer(ImmutableObject):
-
     def __init__(self, impl):
         self.impl = impl
 
-    def visualize(self, U, discretization, title=None, legend=None, separate_colorbars=True, filename=None):
+    def visualize(
+        self,
+        U,
+        discretization,
+        title=None,
+        legend=None,
+        separate_colorbars=True,
+        filename=None,
+    ):
         if isinstance(U, VectorArray):
             U = [U]
         if any(len(u) != 1 for u in U):
@@ -21,12 +28,12 @@ def visualize(self, U, discretization, title=None, legend=None, separate_colorba
         assert title is None or filename is None
         assert legend is None or len(legend) == len(U)
 
-        base_name = title or filename or 'out'
+        base_name = title or filename or "out"
         if len(U) == 1 and not legend:
             filenames = [base_name]
         else:
             legend = legend or list(map(str, list(range(len(U)))))
-            filenames = ['_'.join((base_name, l)) for l in legend]
+            filenames = ["_".join((base_name, l)) for l in legend]
 
         for u, n in zip(U, filenames):
-            self.impl.visualize(u._list[0].impl, n + '.vtk')
+            self.impl.visualize(u._list[0].impl, n + ".vtk")

src/pymor_dealii/pymor/operator.py

@@ -43,7 +43,9 @@ def apply_inverse(self, V, mu=None, least_squares=False):
         return R
 
     def assemble_lincomb(self, operators, coefficients, solver_options=None, name=None):
-        if not all(isinstance(op, (DealIIMatrixOperator, ZeroOperator)) for op in operators):
+        if not all(
+            isinstance(op, (DealIIMatrixOperator, ZeroOperator)) for op in operators
+        ):
             return None
         assert not solver_options  # linear solver is not yet configurable
 

src/pymor_dealii/pymor/vectorarray.py

@@ -4,6 +4,7 @@
 
 try:
     import pymor_dealii_bindings as pd2
+
     HAVE_DEALII = True
 except ImportError:
     HAVE_DEALII = False
@@ -46,7 +47,7 @@ def _axpy(self, alpha, x):
         if x.dim == 0:
             return
         if x is self:
-            self.impl *= 1. + alpha
+            self.impl *= 1.0 + alpha
         else:
             self.impl.axpy(alpha, x.impl)
 
@@ -61,7 +62,7 @@ def norm(self):
         return 0 if self.dim == 0 else self.impl.l2_norm()
 
     def norm2(self):
-        return 0 if self.dim == 0 else self.impl.l2_norm()**2
+        return 0 if self.dim == 0 else self.impl.l2_norm() ** 2
 
     def sup_norm(self):
         return 0 if self.dim == 0 else self.impl.linfty_norm()
@@ -103,12 +104,15 @@ def __neg__(self):
 
 
 class DealIIVectorSpace(ListVectorSpace):
-
     def __init__(self, dim, id=None):
         self.__auto_init(locals())
 
     def __eq__(self, other):
-        return type(other) is DealIIVectorSpace and self.dim == other.dim and self.id == other.id
+        return (
+            type(other) is DealIIVectorSpace
+            and self.dim == other.dim
+            and self.id == other.id
+        )
 
     @classmethod
     def space_from_vector_obj(cls, vec, id):