def test_empty(self):
inds, vals = write_utils.build_ind_val_matrices([[0]],
is_spectral=True)
self.assertTrue(
np.allclose(
inds,
write_utils.INDICES_DTYPE(np.expand_dims(np.arange(1), 0))))
self.assertTrue(
np.allclose(
vals, write_utils.VALUES_DTYPE(np.expand_dims(np.arange(1),
0))))
def test_1D_pos(self):
sine_val = np.sin(np.linspace(0, 2 * np.pi, 128))
inds, vals = write_utils.build_ind_val_matrices([sine_val],
is_spectral=False)
self.assertTrue(
np.allclose(
inds,
write_utils.INDICES_DTYPE(
np.expand_dims(np.arange(len(sine_val)), axis=1))))
self.assertTrue(
np.allclose(
vals, write_utils.VALUES_DTYPE(np.expand_dims(sine_val,
axis=1))))
