|
def test_csv_io(self): |
|
a = tm.TransitionMatrix() |
|
a.to_csv("test.csv") |
|
b = tm.TransitionMatrix(csv_file="test.csv") |
|
self.assertAlmostEqual(a[0, 0], b[0, 0], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[0, 1], b[0, 1], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[1, 0], b[1, 0], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[1, 1], b[1, 1], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
|
|
def test_json_io(self): |
|
a = tm.TransitionMatrix() |
|
a.to_json("test.json") |
|
b = tm.TransitionMatrix(json_file="test.json") |
|
self.assertAlmostEqual(a[0, 0], b[0, 0], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[0, 1], b[0, 1], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[1, 0], b[1, 0], places=ACCURATE_DIGITS, msg=None, delta=None) |
|
self.assertAlmostEqual(a[1, 1], b[1, 1], places=ACCURATE_DIGITS, msg=None, delta=None) |
transitionMatrix/tests/test_model.py
Lines 42 to 58 in 95fca15