snek-ipy-load: also add np and plt variables

snek5000 · Nov 18, 2022 · c62b635 · c62b635
1 parent 6c84fd1
commit c62b635
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/snek5000/util/console.py b/src/snek5000/util/console.py
@@ -36,11 +36,13 @@ def start_ipython_load_sim():
     argv = ["--matplotlib", "-i", "-c"]
     code = dedent(
         """
+        import numpy as np
+        import matplotlib.pyplot as plt
         from fluidsim import load
         print("Loading simulation")
         sim = load()
         params = sim.params
-        print("`sim` and `params` variables are available")
+        print("`sim`, `params`, `np` and `plt` variables are available")
     """
     )
     argv.append("; ".join(code.strip().split("\n")))