def test_run_end_block():
with eventio.IACTFile(testfile) as f:
# looping through all events will fill the run_end block in the end
for event in f:
pass
assert hasattr(f, 'run_end')
def test_next_iter_zstd():
importorskip('zstandard')
# zstd does not support backwards seeking
with raises(ValueError):
with eventio.IACTFile(testfile_zstd) as f:
next(iter(f))
next(iter(f))
def test_bunches_2():
columns = ('x', 'y', 'cx', 'cy', 'time', 'zem', 'photons', 'wavelength')
with eventio.IACTFile(testfile_two_telescopes) as f:
for event in f:
assert len(event.photon_bunches) == 2
assert event.photon_bunches[1].dtype.names == columns
def test_iterating_zstd():
importorskip('zstandard')
with eventio.IACTFile(testfile_zstd) as f:
for i, event in enumerate(f):
assert isinstance(event, eventio.iact.Event)
if i > 5:
break
def test_n_events():
''' tests if the number of events in the testfile is 1 '''
with eventio.IACTFile(testfile) as f:
for i, event in enumerate(f):
pass
assert i == 0
assert event.event_number == 1
assert event.reuse == 1
def test_event_header():
with eventio.IACTFile(testfile) as f:
event = next(iter(f))
assert hasattr(event, 'header')
assert event.header['event_number'] == 1
assert event.header['zenith'] == approx(0.0)
assert event.header['azimuth'] == approx(0.0)
assert event.header['total_energy'] == approx(9.3249321)
def test_bunches():
with eventio.IACTFile(testfile) as f:
event = next(iter(f))
columns = ('x', 'y', 'cx', 'cy', 'time', 'zem', 'photons', 'lambda',
'scattered')
assert event.photon_bunches[0].shape == (382, )
assert event.photon_bunches[0].dtype.names == columns
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from pkg_resources import resource_filename
from argparse import ArgumentParser
import eventio
parser = ArgumentParser()
parser.add_argument('-i', '--inputfile', dest='inputfile')
parser.add_argument('-e', '--event', dest='event', type=int, default=0)
args = parser.parse_args()
testfile = resource_filename('eventio', 'resources/3_gammas_reuse_5.dat')
with eventio.IACTFile(args.inputfile or testfile) as f:
it = iter(f)
event = next(it)
for i in range(args.event):
event = next(it)
fig = plt.figure()
ax = fig.add_axes([0, 0, 1, 1], projection='3d')
obslevel = event.header.observation_levels[0]
for pos, b in zip(f.telescope_positions, event.photon_bunches.values()):
cz = 1 - (b['cx']**2 + b['cy']**2)
x = b['x'] + ((b['zem'] - obslevel) / cz) * b['cx']
y = b['y'] + ((b['zem'] - obslevel) / cz) * b['cy']
def test_next_iter():
with eventio.IACTFile(testfile) as f:
first = next(iter(f))
second = next(iter(f))
assert first.header['event_number'] == second.header['event_number']
def test_iter():
importorskip('zstandard')
with eventio.IACTFile(testfile_zstd) as f:
it = iter(f)
for i in range(10):
next(it)
def test_iterating():
with eventio.IACTFile(testfile) as f:
for event in f:
assert isinstance(event, eventio.iact.Event)
def test_read_telescopes_2():
with eventio.IACTFile(testfile_two_telescopes) as f:
assert f.n_telescopes == 2
assert hasattr(f, 'telescope_positions')
assert f.telescope_positions['x'][1] == approx(5000)
def test_read_telescopes():
with eventio.IACTFile(testfile) as f:
assert f.n_telescopes == 1
assert hasattr(f, 'telescope_positions')
assert f.telescope_positions['x'][0] == approx(0)
def test_read_input_card():
with eventio.IACTFile(testfile) as f:
assert hasattr(f, 'input_card')
def test_read_run_header():
with eventio.IACTFile(testfile) as f:
assert hasattr(f, 'header')
assert f.header['energy_min'] == approx(5.0)
assert f.header['energy_max'] == approx(100.0)
assert f.header['energy_spectrum_slope'] == approx(-2.7)
def test_event_with_reuse():
with eventio.IACTFile(testfile_reuse) as f:
for i, e in enumerate(f):
assert e.event_number == i // 5 + 1
assert e.reuse == (i % 5) + 1
def test_file_open():
eventio.IACTFile(testfile)
