def test_defaults_extended(self):
"""run it and check that defaults were set correctly (extended_out)"""
a = spectrogram(self.data,
variables=self.kwargs['variables'],
extended_out=True)
ans = {
'bins': [
dm.dmarray([
0.00120857, 0.07751865, 0.15382872, 0.2301388, 0.30644887,
0.38275895, 0.45906902, 0.5353791, 0.61168917, 0.68799925,
0.76430932, 0.8406194, 0.91692947, 0.99323955
]),
dm.dmarray([
0.00169679, 0.07848775, 0.1552787, 0.23206965, 0.30886061,
0.38565156, 0.46244251, 0.53923347, 0.61602442, 0.69281538,
0.76960633, 0.84639728, 0.92318824, 0.99997919
])
],
'variables': ['xval', 'yval', 'zval'],
'xlim': (0.0012085702179961411, 0.99323954710300699),
'ylim': (0.001696792515639145, 0.99997919064162388),
'zlim': (0.012544022260691956, 0.99059103521121727)
}
for key in ans:
if key == 'variables':
self.assertEqual(a.specSettings[key], ans[key])
else:
# np.testing.assert_allclose(a.specSettings[key], ans[key], rtol=1e-5)
np.testing.assert_almost_equal(a.specSettings[key],
ans[key],
decimal=8)
def test_defaults(self):
"""run it and check that defaults were set correctly"""
a = spectrogram(self.data, variables=self.kwargs['variables'])
ans = {'bins': [dm.dmarray([ 730120.0, 730135.30769231, 730150.61538462,
730165.92307692, 730181.23076923, 730196.53846154,
730211.84615385, 730227.15384615, 730242.46153846,
730257.76923077, 730273.07692308, 730288.38461538,
730303.69230769, 730319. ]),
dm.dmarray([ 0.00169679, 0.07848775, 0.1552787 , 0.23206965, 0.30886061,
0.38565156, 0.46244251, 0.53923347, 0.61602442, 0.69281538,
0.76960633, 0.84639728, 0.92318824, 0.99997919])],
'variables': ['xval', 'yval', 'zval'],
'ylim': (0.0012085702179961411, 0.99323954710300699),
'zlim': (0.001696792515639145, 0.99997919064162388)}
for key in ans:
if key == 'variables':
self.assertEqual(a.specSettings[key], ans[key])
else:
if key == 'bins':
# np.testing.assert_allclose(a.specSettings[key], ans[key], atol=1e-2, rtol=1e-3)
np.testing.assert_almost_equal(a.specSettings[key], ans[key], decimal=2)
else:
# np.testing.assert_allclose(a.specSettings[key], ans[key], rtol=1e-5)
np.testing.assert_almost_equal(a.specSettings[key], ans[key], decimal=6)
self.assertRaises(NotImplementedError, a.add_data, self.data)
def test_add_data(self):
"""run it and check that add_data correctly"""
data = dm.SpaceData(xval = dm.dmarray(np.arange(3)),
yval = dm.dmarray(np.arange(3)),
zval = dm.dmarray(np.arange(3)))
xbins = np.arange(-0.5, 3.5, 2.0)
ybins = np.arange(-0.5, 3.5, 2.0)
a = spectrogram(self.data, variables=self.kwargs['variables'], extended_out=True)
count = a['spectrogram']['count'][:].copy()
sm = a['spectrogram']['sum'][:].copy()
spect = a['spectrogram']['spectrogram'][:].copy()
a.add_data(self.data) # add te same data back, sum, count will double, spectrogram stays the same
np.testing.assert_almost_equal(a['spectrogram']['count'].filled(), (count*2).filled())
np.testing.assert_almost_equal(a['spectrogram']['sum'], sm*2)
np.testing.assert_almost_equal(a['spectrogram']['spectrogram'], spect)
def test_defaults_extended(self):
"""run it and check that defaults were set correctly (extended_out)"""
a = spectrogram(self.data, variables=self.kwargs['variables'])
ans = {'bins': [dm.dmarray([ 0.00120857, 0.07751865, 0.15382872, 0.2301388 , 0.30644887,
0.38275895, 0.45906902, 0.5353791 , 0.61168917, 0.68799925,
0.76430932, 0.8406194 , 0.91692947, 0.99323955]),
dm.dmarray([ 0.00169679, 0.07848775, 0.1552787 , 0.23206965, 0.30886061,
0.38565156, 0.46244251, 0.53923347, 0.61602442, 0.69281538,
0.76960633, 0.84639728, 0.92318824, 0.99997919])],
'variables': ['xval', 'yval', 'zval'],
'xlim': (0.0012085702179961411, 0.99323954710300699),
'ylim': (0.001696792515639145, 0.99997919064162388),
'zlim': (0.012544022260691956, 0.99059103521121727)}
for key in ans:
if key == 'variables':
self.assertEqual(a.specSettings[key], ans[key])
else:
# np.testing.assert_allclose(a.specSettings[key], ans[key], rtol=1e-5)
np.testing.assert_almost_equal(a.specSettings[key], ans[key], decimal=8)
def test_add_data(self):
"""run it and check that add_data correctly"""
data = dm.SpaceData(xval=dm.dmarray(np.arange(3)),
yval=dm.dmarray(np.arange(3)),
zval=dm.dmarray(np.arange(3)))
xbins = np.arange(-0.5, 3.5, 2.0)
ybins = np.arange(-0.5, 3.5, 2.0)
a = spectrogram(self.data,
variables=self.kwargs['variables'],
extended_out=True)
count = a['spectrogram']['count'][:].copy()
sm = a['spectrogram']['sum'][:].copy()
spect = a['spectrogram']['spectrogram'][:].copy()
a.add_data(
self.data
) # add te same data back, sum, count will double, spectrogram stays the same
np.testing.assert_almost_equal(a['spectrogram']['count'].filled(),
(count * 2).filled())
np.testing.assert_almost_equal(a['spectrogram']['sum'], sm * 2)
np.testing.assert_almost_equal(a['spectrogram']['spectrogram'], spect)
def test_defaults(self):
"""run it and check that defaults were set correctly"""
a = spectrogram(self.data, variables=self.kwargs['variables'])
ans = {
'bins': [
dm.dmarray([
730120.0, 730135.30769231, 730150.61538462,
730165.92307692, 730181.23076923, 730196.53846154,
730211.84615385, 730227.15384615, 730242.46153846,
730257.76923077, 730273.07692308, 730288.38461538,
730303.69230769, 730319.
]),
dm.dmarray([
0.00169679, 0.07848775, 0.1552787, 0.23206965, 0.30886061,
0.38565156, 0.46244251, 0.53923347, 0.61602442, 0.69281538,
0.76960633, 0.84639728, 0.92318824, 0.99997919
])
],
'variables': ['xval', 'yval', 'zval'],
'ylim': (0.0012085702179961411, 0.99323954710300699),
'zlim': (0.001696792515639145, 0.99997919064162388)
}
for key in ans:
if key == 'variables':
self.assertEqual(a.specSettings[key], ans[key])
else:
if key == 'bins':
# np.testing.assert_allclose(a.specSettings[key], ans[key], atol=1e-2, rtol=1e-3)
np.testing.assert_almost_equal(a.specSettings[key],
ans[key],
decimal=2)
else:
# np.testing.assert_allclose(a.specSettings[key], ans[key], rtol=1e-5)
np.testing.assert_almost_equal(a.specSettings[key],
ans[key],
decimal=6)
self.assertRaises(NotImplementedError, a.add_data, self.data)
