Black, black, black..

Author: lisawim

pySDC/projects/PinTSimE/battery_model.py

@@ -247,6 +247,7 @@ def plot_voltages(description, problem, sweeper, recomputed, use_switch_estimato
     fig.savefig('data/{}_model_solution_{}.png'.format(problem, sweeper), dpi=300, bbox_inches='tight')
     plt_helper.plt.close(fig)
 
+
 def get_recomputed(stats, type, sortby):
     """
     Function that filters statistics after a recomputation

pySDC/projects/PinTSimE/estimation_check.py

@@ -196,7 +196,7 @@ def check(cwd='./'):
     """
 
     V_ref = 1.0
-    dt_list = [1e-2, 1e-3, 1e-4]
+    dt_list = [4e-2, 4e-3]
     use_switch_estimator = [True, False]
     use_adaptivity = [True, False]
     restarts_SE = []
@@ -220,7 +220,9 @@ def check(cwd='./'):
                     )
 
                     if use_SE:
-                        assert len(get_recomputed(stats, type='switch', sortby='time')) >= 1, 'No switches found for dt={}!'.format(dt_item)
+                        assert (
+                            len(get_recomputed(stats, type='switch', sortby='time')) >= 1
+                        ), 'No switches found for dt={}!'.format(dt_item)
 
                     fname = 'data/battery_dt{}_USE{}_USA{}_{}.dat'.format(dt_item, use_SE, use_A, sweeper.__name__)
                     f = open(fname, 'wb')

pySDC/projects/PinTSimE/estimation_check_extended.py

@@ -130,7 +130,9 @@ def check(cwd='./'):
             stats, description = run(dt=dt_item, use_switch_estimator=use_SE)
 
             if use_SE:
-                assert len(get_recomputed(stats, type='switch', sortby='time')) >= 1, 'No switches found for dt={}!'.format(dt_item)
+                assert (
+                    len(get_recomputed(stats, type='switch', sortby='time')) >= 1
+                ), 'No switches found for dt={}!'.format(dt_item)
 
             fname = 'data/battery_2condensators_dt{}_USE{}.dat'.format(dt_item, use_SE)
             f = open(fname, 'wb')

pySDC/projects/PinTSimE/switch_estimator.py

@@ -64,9 +64,7 @@ def get_new_step_size(self, controller, S):
             self.params.switch_detected, m_guess, vC_switch = L.prob.get_switching_info(L.u, L.time)
 
             if self.params.switch_detected:
-                t_interp = [
-                    L.time + L.dt * self.params.coll_nodes[m] for m in range(len(self.params.coll_nodes))
-                ]
+                t_interp = [L.time + L.dt * self.params.coll_nodes[m] for m in range(len(self.params.coll_nodes))]
 
                 # only find root if vc_switch[0], vC_switch[-1] have opposite signs (intermediate value theorem)
                 if vC_switch[0] * vC_switch[-1] < 0: