def test_resolution_dense(self):
#- dense with no offsets specified
data = np.random.uniform(size=(10, 10))
R = Resolution(data)
Rdense = R.todense()
self.assertTrue(np.all(Rdense == data))
#- with offsets
offsets = np.arange(-2, 4)
R = Resolution(data, offsets)
Rdense = R.todense()
for i in offsets:
self.assertTrue(np.all(Rdense.diagonal(i) == data.diagonal(i)), \
"diagonal {} doesn't match".format(i))
#- dense without offsets but larger than default_ndiag
ndiag = desispec.resolution.default_ndiag + 5
data = np.random.uniform(size=(ndiag, ndiag))
Rdense = Resolution(data).todense()
for i in range(ndiag):
if i <= desispec.resolution.default_ndiag // 2:
self.assertTrue(np.all(Rdense.diagonal(i) == data.diagonal(i)), \
"diagonal {} doesn't match".format(i))
self.assertTrue(np.all(Rdense.diagonal(-i) == data.diagonal(-i)), \
"diagonal {} doesn't match".format(-i))
else:
self.assertTrue(np.all(Rdense.diagonal(i) == 0.0), \
"diagonal {} not 0s".format(i))
self.assertTrue(np.all(Rdense.diagonal(-i) == 0.0), \
"diagonal {} not 0s".format(-i))
def test_resolution_dense(self):
#- dense with no offsets specified
data = np.random.uniform(size=(10,10))
R = Resolution(data)
Rdense = R.todense()
self.assertTrue(np.all(Rdense == data))
#- with offsets
offsets = np.arange(-2,4)
R = Resolution(data, offsets)
Rdense = R.todense()
for i in offsets:
self.assertTrue(np.all(Rdense.diagonal(i) == data.diagonal(i)), \
"diagonal {} doesn't match".format(i))
#- dense without offsets but larger than default_ndiag
ndiag = desispec.resolution.default_ndiag + 5
data = np.random.uniform(size=(ndiag, ndiag))
Rdense = Resolution(data).todense()
for i in range(ndiag):
if i <= desispec.resolution.default_ndiag//2:
self.assertTrue(np.all(Rdense.diagonal(i) == data.diagonal(i)), \
"diagonal {} doesn't match".format(i))
self.assertTrue(np.all(Rdense.diagonal(-i) == data.diagonal(-i)), \
"diagonal {} doesn't match".format(-i))
else:
self.assertTrue(np.all(Rdense.diagonal(i) == 0.0), \
"diagonal {} not 0s".format(i))
self.assertTrue(np.all(Rdense.diagonal(-i) == 0.0), \
"diagonal {} not 0s".format(-i))
