refactor tests

Author: ben9809

tests/test_metrics.py

@@ -428,6 +428,7 @@ def test_membrane_potential_updates(self):
 
         out = model(inp)
         tot_mem_updates = self.mem_updates(model, out, (inp, 0))
+        self.mem_updates.reset()
 
         self.assertEqual(tot_mem_updates, 50)
 
@@ -441,6 +442,7 @@ def test_activation_sparsity_by_layer(self):
 
         out = model(inp)
         act_sparsity_by_layer = self.activation_sparsity_by_layer(model, out, (inp, 0))
+        self.activation_sparsity_by_layer.reset()
 
         self.assertEqual(act_sparsity_by_layer["1"], 0.96)