/usr/lib/python3/dist-packages/fabio/test/testhdf5image.py is in python3-fabio 0.6.0+dfsg-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 102 103 104 105 106 | #!/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 sys
import os
if __name__ == '__main__':
import pkgutil
__path__ = pkgutil.extend_path([os.path.dirname(__file__)], "fabio.test")
from .utilstest import UtilsTest
logger = UtilsTest.get_logger(__file__)
fabio = sys.modules["fabio"]
from fabio.fabioutils import exists
from fabio.openimage import openimage
from fabio.hdf5image import Hdf5Image, h5py
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())
|