class SaveTests(unittest.TestCase):
def setUp(self):
self.s = Signal()
self.x = np.array([0, 1, 2, 3])
self.y = np.array([0, 1, 0, -1])
self.s.load_nparray([0, 1, 0, -1], 'signal', 'm', 1)
self.s.fft()
self.s.freq_filter_Hilbert_complex()
self.s.ifft()
self.s.f.attrs['two'] = 2 # test attribute to verify attrs copied
self.f_dst = h5py.File('.test2.h5', backing_store=False, driver='core')
def test_save_pass_list_datasets(self):
self.s.save(self.f_dst, ['x', 'y'])
assert_array_equal(self.f_dst['x'][:], self.x)
assert_array_equal(self.f_dst['y'][:], self.y)
self.assertEqual(self.f_dst.attrs['two'], 2)
def test_save_pass_string(self):
self.s.save(self.f_dst, 'input')
assert_array_equal(self.f_dst['x'][:], self.x[:])
assert_array_equal(self.f_dst['y'][:], self.y)
self.assertEqual(self.f_dst.attrs['two'], 2)
def tearDown(self):
self.f_dst.close()
self.s.close()
class SaveTests(unittest.TestCase):
def setUp(self):
self.s = Signal()
self.x = np.array([0, 1, 2, 3])
self.y = np.array([0, 1, 0, -1])
self.s.load_nparray([0, 1, 0, -1], 'signal', 'm', 1)
self.s.fft()
self.s.freq_filter_Hilbert_complex()
self.s.ifft()
self.s.f.attrs['two'] = 2 # test attribute to verify attrs copied
self.f_dst = h5py.File('.test2.h5', backing_store=False, driver='core')
def test_save_pass_list_datasets(self):
self.s.save(self.f_dst, ['x', 'y'])
assert_array_equal(self.f_dst['x'][:], self.x)
assert_array_equal(self.f_dst['y'][:], self.y)
self.assertEqual(self.f_dst.attrs['two'], 2)
def test_save_pass_string(self):
self.s.save(self.f_dst, 'input')
assert_array_equal(self.f_dst['x'][:], self.x[:])
assert_array_equal(self.f_dst['y'][:], self.y)
self.assertEqual(self.f_dst.attrs['two'], 2)
def tearDown(self):
self.f_dst.close()
self.s.close()
class SaveBeforeWorkupTest(unittest.TestCase):
def setUp(self):
self.s = Signal()
self.x = np.array([0, 1, 2, 3])
self.y = np.array([0, 1, 0, -1])
self.s.load_nparray([0, 1, 0, -1], 'signal', 'm', 1)
self.f_dst = h5py.File('.test3.h5', backing_store=False, driver='core')
def test_save_before_workup(self):
self.s.save(self.f_dst, 'time_workup')
assert_array_equal(self.f_dst['x'][:], self.x)
assert_array_equal(self.f_dst['y'][:], self.y)
def tearDown(self):
self.f_dst.close()
self.s.close()
class SaveBeforeWorkupTest(unittest.TestCase):
def setUp(self):
self.s = Signal()
self.x = np.array([0, 1, 2, 3])
self.y = np.array([0, 1, 0, -1])
self.s.load_nparray([0, 1, 0, -1], 'signal', 'm', 1)
self.f_dst = h5py.File('.test3.h5', backing_store=False, driver='core')
def test_save_before_workup(self):
self.s.save(self.f_dst, 'time_workup')
assert_array_equal(self.f_dst['x'][:], self.x)
assert_array_equal(self.f_dst['y'][:], self.y)
def tearDown(self):
self.f_dst.close()
self.s.close()
