diff --git a/galvani/BioLogic.py b/galvani/BioLogic.py index 0fa07f0..a34ad13 100644 --- a/galvani/BioLogic.py +++ b/galvani/BioLogic.py @@ -70,7 +70,8 @@ def MPTfile(file_or_path): if magic != b'EC-Lab ASCII FILE\r\n': raise ValueError("Bad first line for EC-Lab file: '%s'" % magic) - nb_headers_match = re.match(b'Nb header lines : (\d+)\s*$', next(mpt_file)) + # TODO use rb'string' here once Python 2 is no longer supported + nb_headers_match = re.match(b'Nb header lines : (\\d+)\\s*$', next(mpt_file)) nb_headers = int(nb_headers_match.group(1)) if nb_headers < 3: raise ValueError("Too few header lines: %d" % nb_headers) @@ -107,7 +108,7 @@ def MPTfileCSV(file_or_path): if magic.rstrip() != 'EC-Lab ASCII FILE': raise ValueError("Bad first line for EC-Lab file: '%s'" % magic) - nb_headers_match = re.match('Nb header lines : (\d+)\s*$', next(mpt_file)) + nb_headers_match = re.match(r'Nb header lines : (\d+)\s*$', next(mpt_file)) nb_headers = int(nb_headers_match.group(1)) if nb_headers < 3: raise ValueError("Too few header lines: %d" % nb_headers) @@ -275,7 +276,7 @@ def read_VMP_modules(fileobj, read_module_data=True): if len(hdr_bytes) < VMPmodule_hdr.itemsize: raise IOError("Unexpected end of file while reading module header") - hdr = np.fromstring(hdr_bytes, dtype=VMPmodule_hdr, count=1) + hdr = np.frombuffer(hdr_bytes, dtype=VMPmodule_hdr, count=1) hdr_dict = dict(((n, hdr[n][0]) for n in VMPmodule_hdr.names)) hdr_dict['offset'] = fileobj.tell() if read_module_data: @@ -326,17 +327,16 @@ class MPRfile: data_module, = (m for m in modules if m['shortname'] == b'VMP data ') maybe_log_module = [m for m in modules if m['shortname'] == b'VMP LOG '] - n_data_points = np.fromstring(data_module['data'][:4], dtype=' 40000 and ole_timestamp1 < 50000: