Merge pull request #132 from fact-project/fix_reuse_add_test

Fix reuse add test

Author: maxnoe

fact/VERSION

@@ -1 +1 @@
-0.25.1
+0.25.2

fact/io.py

@@ -10,7 +10,6 @@
 import astropy.units as u
 import tempfile
 import warnings
-from time import perf_counter
 
 
 __all__ = [
@@ -559,7 +558,7 @@ def read_simulated_spectrum(corsika_headers_path):
         if 'n_reuse' in runs.columns:
             # if reuse is not the same for all runs, multply n_showers
             # and set reuse to 1
-            unique = runs['reuse'].unique()
+            unique = runs['n_reuse'].unique()
             if len(unique) > 1:
                 summary['n_showers'] = (n_showers * runs['n_reuse']).sum()
                 summary['n_reuse'] = 1

tests/resources/proton_header_test.hdf5

@@ -1,10 +1,9 @@
-import tables.filters
 import pandas as pd
 import tempfile
 import numpy as np
 import h5py
 import pytest
-from pandas.util.testing import assert_frame_equal
+
 
 
 def test_to_h5py():
@@ -335,3 +334,16 @@ def test_compression():
         assert all(df.dtypes == df2.dtypes)
         assert all(df['x'] == df2['x'])
         assert all(df['N'] == df2['N'])
+
+
+def test_read_simulated_spectrum():
+    from fact.io import read_simulated_spectrum
+    import astropy.units as u
+
+    s = read_simulated_spectrum('tests/resources/proton_header_test.hdf5')
+
+    assert s['n_showers'] == 20000
+    assert s['n_reuse'] == 20
+    assert s['energy_min'] == 100 * u.GeV
+    assert s['energy_max'] == 200 * u.TeV
+    assert s['energy_spectrum_slope'] == -2.0