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)