diff --git a/bfps/DNS.py b/bfps/DNS.py index b17d2611a13b363dc4740b0ae0d2f0e4fc6084b6..f24cd59b8d2edd3c6460996cac9c9bb7f40eca00 100644 --- a/bfps/DNS.py +++ b/bfps/DNS.py @@ -74,6 +74,7 @@ class DNS(_code): '"scope_timer.hpp"', '"fftw_interface.hpp"', '"full_code/main_code.hpp"', + '<cmath>', '<iostream>', '<hdf5.h>', '<string>', diff --git a/tests/test_DNS.py b/tests/test_DNS.py index 3ff6596330e2192bb919dd9611baff03f56fd473..bac769522743232120dea26a55e0fa2fedbb480f 100644 --- a/tests/test_DNS.py +++ b/tests/test_DNS.py @@ -2,7 +2,7 @@ from bfps.DNS import DNS def main(): - c = DNS() + c = DNS(dns_type = 'NSVEp') c.write_src() c.compile_code() return None