Files
galvani/tests/test_BioLogic.py

91 lines
3.3 KiB
Python

# -*- coding: utf-8 -*-
import os.path
from datetime import date
import numpy as np
from numpy.testing import assert_array_almost_equal, assert_array_equal
from nose.tools import ok_, eq_, raises
from ..import MPTfile, MPRfile
from ..BioLogic import MPTfileCSV # not exported
testdata_dir = os.path.join(os.path.dirname(__file__), 'testdata')
def test_open_MPT():
mpt1, comments = MPTfile(os.path.join(testdata_dir, 'bio-logic1.mpt'))
eq_(comments, [])
eq_(mpt1.dtype.names, ("mode", "ox/red", "error", "control changes",
"Ns changes", "counter inc.", "time/s",
"control/V/mA", "Ewe/V", "dQ/mA.h", "P/W",
"I/mA", "(Q-Qo)/mA.h", "x"))
@raises(ValueError)
def test_open_MPT_fails_for_bad_file():
mpt1 = MPTfile(os.path.join(testdata_dir, 'bio-logic1.mpr'))
def test_open_MPT_csv():
mpt1, comments = MPTfileCSV(os.path.join(testdata_dir, 'bio-logic1.mpt'))
eq_(comments, [])
eq_(mpt1.fieldnames, ["mode", "ox/red", "error", "control changes",
"Ns changes", "counter inc.", "time/s",
"control/V/mA", "Ewe/V", "dq/mA.h", "P/W",
"<I>/mA", "(Q-Qo)/mA.h", "x"])
@raises(ValueError)
def test_open_MPT_csv_fails_for_bad_file():
mpt1 = MPTfileCSV(os.path.join(testdata_dir, 'bio-logic1.mpr'))
def test_MPR_read_dates():
mpr1 = MPRfile(os.path.join(testdata_dir, 'bio-logic1.mpr'))
eq_(mpr1.startdate, date(2011, 10, 29))
eq_(mpr1.enddate, date(2011, 10, 31))
@raises(ValueError)
def test_open_MPR_fails_for_bad_file():
mpr1 = MPRfile(os.path.join(testdata_dir, 'arbin1.res'))
def test_MPR_matches_MPT():
mpr1 = MPRfile(os.path.join(testdata_dir, 'bio-logic1.mpr'))
mpt1, comments = MPTfile(os.path.join(testdata_dir, 'bio-logic1.mpt'))
assert_array_equal(mpr1.data["flags"] & 0x03, mpt1["mode"])
assert_array_equal(np.array(mpr1.data["flags"] & 0x04, dtype=np.bool_),
mpt1["ox/red"])
assert_array_equal(np.array(mpr1.data["flags"] & 0x08, dtype=np.bool_),
mpt1["error"])
assert_array_equal(np.array(mpr1.data["flags"] & 0x10, dtype=np.bool_),
mpt1["control changes"])
assert_array_equal(np.array(mpr1.data["flags"] & 0x20, dtype=np.bool_),
mpt1["Ns changes"])
## Nothing uses the 0x40 bit of the flags
assert_array_equal(np.array(mpr1.data["flags"] & 0x80, dtype=np.bool_),
mpt1["counter inc."])
assert_array_almost_equal(mpr1.data["time/s"],
mpt1["time/s"],
decimal=5) # 5 digits in CSV
assert_array_almost_equal(mpr1.data["control/V/mA"],
mpt1["control/V/mA"],
decimal=6) # 32 bit float precision
assert_array_almost_equal(mpr1.data["Ewe/V"],
mpt1["Ewe/V"],
decimal=6) # 32 bit float precision
assert_array_almost_equal(mpr1.data["dQ/mA.h"],
mpt1["dQ/mA.h"],
decimal=17) # 64 bit float precision
assert_array_almost_equal(mpr1.data["P/W"],
mpt1["P/W"],
decimal=10) # 32 bit float precision for 1.xxE-5