def test_bigspec(self):
array1 = numpy.linspace(1000,10000, num=100)
array2 = numpy.arange(100,9900.1, 50)
# check linear binning
result = calc.big_spec(numpy.append(array1,array2), 50, False)
numpy.testing.assert_array_equal(numpy.arange(100,10000+50, 50), result)
self.assertEqual(result[len(result)-1], 10000)
# check log binning
test = calc.big_spec(numpy.append(array1,array2), 0.1, True)
result = 10**numpy.array([2.,2.1,2.2,2.3,2.4,
2.5,2.6,2.7,2.8,2.9,
3.,3.1,3.2,3.3,3.4,
3.5,3.6,3.7,3.8,3.9,
4.])
numpy.testing.assert_array_almost_equal(test, result)
def test_setup_binup_logbinFalse(self):
x = [1234.,3365.,4407.,5657.,4479.0,6535.]
y = [1]*len(x)
yerr = y
logbin = False
binsize = 100
xarr = calc.big_spec(x,binsize,logbin)
x_control = numpy.array(x)
x, xarr, y, m_yerr, nx, xbin, yarr, outerr, count, skipit = calc.setup_binup_arrays(y, x, xarr, binsize, yerr, logbin=logbin)
numpy.testing.assert_array_almost_equal(x, x_control)
self.assertEqual(xbin, 50.0)
self.assertEqual(len(m_yerr), len(x))
def test_setup_binup_logbinTrue(self):
x = [1234.,3365.,4407.,5657.,4479.0,6535.]
y = [1]*len(x)
yerr = y
logbin = True
binsize = 0.1
xarr = calc.big_spec(x,binsize,logbin)
x_control = numpy.log10(x)
x, xarr, y, m_yerr, nx, xbin, yarr, outerr, count, skipit = calc.setup_binup_arrays(y, x, xarr, binsize, yerr, logbin=logbin)
numpy.testing.assert_array_almost_equal(x, x_control)
self.assert_(isinstance(xarr, numpy.ndarray))
self.assertEqual(xbin, 0.05)
self.assertEqual(len(yarr), nx)
