Add check for complex-valued assignment in real-valued GridFunction o…

…bject
awslabs · Mar 4, 2024 · 702eb30 · 702eb30
1 parent 1ec7cb0
commit 702eb30
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/palace/fem/gridfunction.cpp b/palace/fem/gridfunction.cpp
@@ -29,6 +29,12 @@ GridFunction &GridFunction::operator=(std::complex<double> s)
   {
     Imag() = s.imag();
   }
+  else
+  {
+    MFEM_ASSERT(
+        s.imag() == 0.0,
+        "Cannot assign complex scalar to a non-complex-valued GridFunction object!");
+  }
   return *this;
 }