From 02fbf7c9b2313bc77d93c7ea7c085d8af6d9ed42 Mon Sep 17 00:00:00 2001
From: tlatorre <tlatorre@uchicago.edu>
Date: Tue, 18 Sep 2018 18:08:01 -0500
Subject: free muon_energy struct in test_muon_get_energy()

---
 src/test.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/test.c')

diff --git a/src/test.c b/src/test.c
index 6bb2046..e46963b 100644
--- a/src/test.c
+++ b/src/test.c
@@ -125,7 +125,7 @@ int test_muon_get_energy(char *err)
     /* At the beginning of the track we should have roughly the same energy. */
     if (!isclose(T, T0, 1e-5, 0)) {
         sprintf(err, "KE = %.5f, but expected %.5f", T, T0);
-        return 1;
+        goto err;
     }
 
     range = get_range(T0,1.0);
@@ -134,10 +134,18 @@ int test_muon_get_energy(char *err)
     /* At the end of the track the energy should be approximately 0. */
     if (!isclose(T, 0, 1e-5, 1e-5)) {
         sprintf(err, "KE = %.5f, but expected %.5f", T, 0.0);
-        return 1;
+        goto err;
     }
 
+    muon_free_energy(m);
+
     return 0;
+
+err:
+    muon_free_energy(m);
+
+    return 1;
+
 }
 
 int test_muon_get_range(char *err)
-- 
cgit