Update build test and integration testing suite (#1324)

Author: Saransh-cpp

.github/workflows/build.yml

@@ -14,12 +14,12 @@ concurrency:
   cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }}
 
 jobs:
-  build:
+  import:
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
         os: [ubuntu-latest, macos-latest, windows-latest]
 
     steps:

deepxde/config.py

@@ -1,5 +1,6 @@
 import os
 import random
+import sys
 
 import numpy as np
 
@@ -52,7 +53,7 @@
 elif backend_name == "jax":
     xla_jit = True
 if xla_jit:
-    print("Enable just-in-time compilation with XLA.\n")
+    print("Enable just-in-time compilation with XLA.\n", file=sys.stderr, flush=True)
 
 
 def default_float():
@@ -179,9 +180,9 @@ def enable_xla_jit(mode=True):
     global xla_jit
     xla_jit = mode
     if xla_jit:
-        print("Enable just-in-time compilation with XLA.\n")
+        print("Enable just-in-time compilation with XLA.\n", file=sys.stderr, flush=True)
     else:
-        print("Disable just-in-time compilation with XLA.\n")
+        print("Disable just-in-time compilation with XLA.\n", file=sys.stderr, flush=True)
 
 
 def disable_xla_jit():

examples/Makefile

@@ -1,5 +1,5 @@
 # Directories where to look for *.py example scripts
-EXAMPLE_DIRECTORIES := function pinn_forward pinn_inverse
+EXAMPLE_DIRECTORIES := function operator pinn_forward pinn_inverse
 
 # Environment variables for the scripts
 # Note that the PYTHONPATH is relative to each example subdirectory, not the
@@ -24,12 +24,12 @@ run_all_examples: $(example_files)
 	@echo -----------------------------------------------------
 	@echo $@
 	@echo -----------------------------------------------------
-	@cd $(dir $@) && $(DDE_TEST_ENV) python ../sample_to_test.py $(notdir $@) | PYTHONPATH=../.. python -
+	@cd $(dir $@) && $(DDE_TEST_ENV) python3 ../sample_to_test.py $(notdir $@) | PYTHONPATH=../.. python3 -
 	@echo
 	@echo
 	@cd ..
 
 
 # Clean all data files in example directories
-clean: $(foreach dir,$(EXAMPLE_DIRECTORIES),$(wildcard $(dir)/*.dat)) $(foreach dir,$(EXAMPLE_DIRECTORIES),$(wildcard $(dir)/*.npz))
+clean: $(foreach dir,$(EXAMPLE_DIRECTORIES),$(wildcard $(dir)/*.dat)) $(foreach dir,$(EXAMPLE_DIRECTORIES),$(wildcard $(dir)/*.npz)) $(foreach dir,$(EXAMPLE_DIRECTORIES),$(wildcard $(dir)/*.pdf))
 	rm $^

examples/sample_to_test.py

@@ -10,8 +10,10 @@
 import re
 import sys
 
+import deepxde as dde
 
-EPOCHS_RE = re.compile(r"^(.*)epochs\s*=\s*(\d+)(.*)$", re.DOTALL)
+
+ITERATIONS_RE = re.compile(r"^(.*)iterations\s*=\s*(\d+)(.*)$", re.DOTALL)
 
 PROLOG = """
 from deepxde.optimizers import set_LBFGS_options
@@ -24,9 +26,9 @@
 
 def transform(line, file_name):
     """Apply transformations to line."""
-    m = re.match(EPOCHS_RE, line)
+    m = re.match(ITERATIONS_RE, line)
     if m is not None:
-        line = m.expand(r"\1epochs=1\3")
+        line = m.expand(r"\1iterations=1\3")
 
     # Burgers_RAR.py has an additional convergence loop: force 1 single pass
     if file_name == "Burgers_RAR.py" and line.startswith("while"):
@@ -45,5 +47,42 @@ def transform(line, file_name):
 
     print(PROLOG)
     with open(file_name, "r") as input:
-        for line in input:
-            sys.stdout.write(transform(line, file_name))
+        if file_name not in (
+            # this example uses skopt which is not maintained
+            # and uses deprecated numpy API
+            "Allen_Cahn.py",
+            # the below examples have different code for different
+            # backends
+            "Beltrami_flow.py",
+            "Helmholtz_Dirichlet_2d_HPO.py",
+            "Laplace_disk.py",
+            "Lotka_Volterra.py",
+            "Poisson_Dirichlet_1d.py",
+            "Poisson_periodic_1d.py",
+            "diffusion_1d_exactBC.py",
+            "diffusion_1d_resample.py",
+            "diffusion_1d.py",
+            "diffusion_reaction.py",
+            "fractional_diffusion_1d.py",
+            "fractional_Poisson_1d.py",
+            "fractional_Poisson_2d.py",
+            "fractional_Poisson_3d.py",
+            "ide.py",
+            "diffusion_1d_inverse.py",
+            "fractional_Poisson_1d_inverse.py",
+            "fractional_Poisson_2d_inverse.py",
+            "Poisson_Lshape.py",
+            "ode_system.py",
+            "Lorenz_inverse.py",
+            # gives error with tensorflow.compat.v1
+            "Volterra_IDE.py",
+            # the dataset is large and not included in repo
+            "antiderivative_unaligned.py",
+            "antiderivative_aligned.py",
+        ):
+            lines = input.readlines()
+            if dde.backend.get_preferred_backend() in lines[0].replace(",", "").replace(
+                '"""', ""
+            ).replace("\n", "").split(" "):
+                for line in lines:
+                    sys.stdout.write(transform(line, file_name))