Merge pull request #8 from tataratat/ft-custom-poisson

update formatting, add custom poisson 2d

.pre-commit-config.yaml

@@ -27,17 +27,6 @@ repos:
   - id: requirements-txt-fixer
   - id: trailing-whitespace
 
-- repo: https://github.com/asottile/pyupgrade
-  rev: "v3.15.0"
-  hooks:
-  - id: pyupgrade
-    args: [--py36-plus]
-
-- repo: https://github.com/PyCQA/isort
-  rev: "5.12.0"
-  hooks:
-  - id: isort
-
 - repo: https://github.com/psf/black
   rev: "23.11.0"
   hooks:

examples/custom_poisson2d.py

@@ -0,0 +1,7 @@
+import feigen
+
+if __name__ == "__main__":
+    # feigen.log.configure(debug=True)
+
+    b = feigen.CustomPoisson2D()
+    b.start()

feigen/__init__.py

@@ -1,6 +1,7 @@
-from feigen import bspline, comm, log, poisson2d
+from feigen import bspline, comm, custom_poisson2d, log, poisson2d
 from feigen._version import __version__
 from feigen.bspline import BSpline2D
+from feigen.custom_poisson2d import CustomPoisson2D
 from feigen.poisson2d import Poisson2D
 
 __all__ = [
@@ -9,6 +10,8 @@
     "comm",
     "log",
     "poisson2d",
+    "custom_poisson2d",
     "BSpline2D",
     "Poisson2D",
+    "CustomPoisson2D",
 ]

feigen/bspline.py

@@ -294,13 +294,13 @@ def __init__(self, uri, degree=None, ncoeffs=None):  # noqa PLR0915
         self._state = {}
 
         # plotter initialization constants
-        self._config["dim"] = int(2)  # 2D
-        self._config["n_subplots"] = int(
-            3
-        )  # geometry, boundary condition, server response
-        self._config["geometry_plot"] = int(0)
-        self._config["bc_plot"] = int(1)
-        self._config["server_plot"] = int(2)
+        self._config["dim"] = 2  # 2D
+        self._config[
+            "n_subplots"
+        ] = 3  # geometry, boundary condition, server response
+        self._config["geometry_plot"] = 0
+        self._config["bc_plot"] = 1
+        self._config["server_plot"] = 2
 
         # field dim is temporary for now - (1)
         self._config["field_dim"] = 1