1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Project: Fable Input Output
# https://github.com/silx-kit/fabio
#
# Copyright (C) European Synchrotron Radiation Facility, Grenoble, France
#
# Principal author: Jérôme Kieffer (Jerome.Kieffer@ESRF.eu)
#
"""Test Eiger images
"""
from __future__ import print_function, with_statement, division, absolute_import
import unittest
import os
import logging
logger = logging.getLogger(__name__)
from fabio.fabioutils import exists
from fabio.openimage import openimage
from fabio.hdf5image import Hdf5Image, h5py
from .utilstest import UtilsTest
def make_hdf5(name, shape=(50, 99, 101)):
if h5py is None:
raise unittest.SkipTest("h5py is not available")
with h5py.File(name) as h:
e = h.require_group("entry")
if len(shape) == 2:
e.require_dataset("data", shape, compression="gzip", compression_opts=9, dtype="float32")
elif len(shape) == 3:
e.require_dataset("data", shape, chunks=(1,) + shape[1:], compression="gzip", compression_opts=9, dtype="float32")
return name + "::entry/data"
class TestHdf5(unittest.TestCase):
"""basic test"""
@classmethod
def setUpClass(cls):
super(TestHdf5, cls).setUpClass()
cls.fn2 = os.path.join(UtilsTest.tempdir, "eiger2d.h5")
cls.fn2 = make_hdf5(cls.fn2, (99, 101))
cls.fn3 = os.path.join(UtilsTest.tempdir, "eiger3d.h5")
cls.fn3 = make_hdf5(cls.fn3, (50, 99, 101))
@classmethod
def tearDownClass(cls):
super(TestHdf5, cls).tearDownClass()
if exists(cls.fn3):
os.unlink(cls.fn3.split("::")[0])
if exists(cls.fn2):
os.unlink(cls.fn2.split("::")[0])
def test_read(self):
""" check we can read images from Eiger"""
e = Hdf5Image()
e.read(self.fn2)
self.assertEqual(e.dim1, 101, "dim1 OK")
self.assertEqual(e.dim2, 99, "dim2 OK")
self.assertEqual(e.nframes, 1, "nframes OK")
self.assertEqual(e.bpp, 4, "nframes OK")
e = Hdf5Image()
e.read(self.fn3)
self.assertEqual(e.dim1, 101, "dim1 OK")
self.assertEqual(e.dim2, 99, "dim2 OK")
self.assertEqual(e.nframes, 50, "nframes OK")
self.assertEqual(e.bpp, 4, "nframes OK")
def test_open(self):
""" check we can read images from Eiger"""
e = openimage(self.fn2)
self.assertEqual(e.dim1, 101, "dim1 OK")
self.assertEqual(e.dim2, 99, "dim2 OK")
self.assertEqual(e.nframes, 1, "nframes OK")
self.assertEqual(e.bpp, 4, "nframes OK")
e = openimage(self.fn3)
self.assertEqual(e.dim1, 101, "dim1 OK")
self.assertEqual(e.dim2, 99, "dim2 OK")
self.assertEqual(e.nframes, 50, "nframes OK")
self.assertEqual(e.bpp, 4, "nframes OK")
def suite():
loadTests = unittest.defaultTestLoader.loadTestsFromTestCase
testsuite = unittest.TestSuite()
testsuite.addTest(loadTests(TestHdf5))
return testsuite
if __name__ == '__main__':
runner = unittest.TextTestRunner()
runner.run(suite())
|
