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
|
# Copyright (c) 2016, Science and Technology Facilities Council
# This software is distributed under a BSD licence. See LICENSE.txt.
"""
Tests for bzip2mrcfile.py
"""
# Import Python 3 features for future-proofing
from __future__ import (absolute_import, division, print_function,
unicode_literals)
import os
import unittest
from . import test_mrcfile
from mrcfile.bzip2mrcfile import Bzip2MrcFile
class Bzip2MrcFileTest(test_mrcfile.MrcFileTest):
"""Unit tests for bzip2 MRC file I/O.
Note that this test class inherits MrcFileTest to ensure all of the tests
for MrcObject and MrcFile work correctly for the Bzip2MrcFile subclass.
"""
def setUp(self):
# Set up as if for MrcFileTest
super(Bzip2MrcFileTest, self).setUp()
# Replace test MRC files with their gzipped equivalents
self.example_mrc_name = os.path.join(self.test_data, 'EMD-3197.map.bz2')
self.ext_header_mrc_name = os.path.join(self.test_data, 'EMD-3001.map.bz2')
self.fei1_ext_header_mrc_name = os.path.join(self.test_data, 'fei-extended.mrc.bz2')
self.fei2_ext_header_mrc_name = os.path.join(self.test_data, 'epu2.9_example.mrc.bz2')
# Create .bz2 files from .mrc files if necessary
if not os.path.isfile(self.fei1_ext_header_mrc_name):
print("Test data file fei-extended.mrc.bz2 not found. Creating a new copy...")
fei1_ext_header_mrc_name = os.path.join(self.test_data, 'fei-extended.mrc')
import bz2
with open(fei1_ext_header_mrc_name, "rb") as mrc:
with bz2.open(self.fei1_ext_header_mrc_name, 'wb') as bzf:
bzf.write(mrc.read())
if not os.path.isfile(self.fei2_ext_header_mrc_name):
print("Test data file epu2.9_example.mrc.bz2 not found. Creating a new copy...")
fei2_ext_header_mrc_name = os.path.join(self.test_data, 'epu2.9_example.mrc')
import bz2
with open(fei2_ext_header_mrc_name, "rb") as mrc:
with bz2.open(self.fei2_ext_header_mrc_name, 'wb') as bzf:
bzf.write(mrc.read())
# Set the newmrc method to the GzipMrcFile constructor
self.newmrc = Bzip2MrcFile
# Set up parameters so MrcObject tests run on the GzipMrcFile class
obj_mrc_name = os.path.join(self.test_output, 'test_mrcobject.mrc')
self.mrcobject = Bzip2MrcFile(obj_mrc_name, 'w+', overwrite=True)
# Flush and re-read to ensure underlying file is valid bzip2
self.mrcobject.flush()
self.mrcobject._read()
def test_non_mrc_file_is_rejected(self):
"""Override test to change expected error message."""
name = os.path.join(self.test_data, 'emd_3197.png')
with (self.assertRaisesRegex(IOError, '[Ii]nvalid data stream')):
Bzip2MrcFile(name)
def test_non_mrc_file_gives_correct_warnings_in_permissive_mode(self):
"""Override test - permissive mode still can't read non-bzip2 files."""
name = os.path.join(self.test_data, 'emd_3197.png')
with (self.assertRaisesRegex(IOError, '[Ii]nvalid data stream')):
Bzip2MrcFile(name, permissive=True)
def test_repr(self):
"""Override test to change expected repr string."""
with Bzip2MrcFile(self.example_mrc_name) as mrc:
assert repr(mrc) == "Bzip2MrcFile('{0}', mode='r')".format(self.example_mrc_name)
if __name__ == "__main__":
unittest.main()
|
