Adding HeatChargeSimulation samples to the sample script and updater test

Author: momchil-flex

scripts/sample.py

@@ -7,10 +7,11 @@
 
 sys.path.append("tests")
 
-from utils import SIM_FULL
+from utils import SAMPLE_SIMULATIONS
 
-FPREFIX_SIMULATION_SAMPLE = join("tests", "sims", "simulation_sample")
+FPREFIX_SIMULATION_SAMPLE = join("tests", "sims")
 
-# Store a simulation sample as json and hdf5
-SIM_FULL.to_json(FPREFIX_SIMULATION_SAMPLE + ".json")
-SIM_FULL.to_hdf5(FPREFIX_SIMULATION_SAMPLE + ".h5")
+for key, sim in SAMPLE_SIMULATIONS.items():
+    # Store a simulation sample as json and hdf5
+    sim.to_json(join(FPREFIX_SIMULATION_SAMPLE, key) + ".json")
+    sim.to_hdf5(join(FPREFIX_SIMULATION_SAMPLE, key) + ".h5")