python2

Author: kain88-de

testsuite/MDAnalysisTests/formats/test_libdcd.py

@@ -20,6 +20,7 @@
 from collections import namedtuple
 import os
 import string
+import struct
 
 import hypothesis.strategies as strategies
 from hypothesis import example, given
@@ -246,17 +247,17 @@ def test_write_header(tmpdir):
     with open(testfile, 'rb') as fh:
         header_bytes = fh.read()
     # check for magic number
-    assert int.from_bytes(header_bytes[:4], byteorder='little') == 84
+    assert struct.unpack('i', header_bytes[:4])[0] == 84
     # magic number should be written again before remark section
-    assert int.from_bytes(header_bytes[88:92], byteorder='little') == 84
+    assert struct.unpack('i', header_bytes[88:92])[0] == 84
     # length of remark section. We hard code this to 244 right now
-    assert int.from_bytes(header_bytes[92:96], byteorder='little') == 244
+    assert struct.unpack('i', header_bytes[92:96])[0] == 244
     # say we have 3 block of length 80
-    assert int.from_bytes(header_bytes[96:100], byteorder='little') == 3
+    assert struct.unpack('i', header_bytes[96:100])[0] == 3
     # after the remark section the length should be reported again
-    assert int.from_bytes(header_bytes[340:344], byteorder='little') == 244
+    assert struct.unpack('i', header_bytes[340:344])[0] == 244
     # this is a magic number as far as I see
-    assert int.from_bytes(header_bytes[344:348], byteorder='little') == 4
+    assert struct.unpack('i', header_bytes[344:348])[0] == 4
 
 
 def test_write_no_header(tmpdir):