Esempio n. 1
0
def test_get_column() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    A = nt.column('A')
    ntl.assert_true(A == range(100))
    B = nt.column('B')
    ntl.assert_true(B == [ x*100 for x in range(100)])
Esempio n. 2
0
def test_get_with_cross_cut() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    col = nt.column('B',lambda x : x.A > 50 and x.B < 8500 )
    ntl.assert_true(col == [x*100 for x in xrange(51, 85)])
Esempio n. 3
0
def test_get_with_cut() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    col = nt.column('A',lambda x : x.A > 50 )
    ntl.assert_true(col == range(51, 100))
Esempio n. 4
0
def reference_ntuple():
    nt = NTuple('x', 'sin', 'cos')
    for x in xrange(-50, 51):
        nt.fill('x', x)
        nt.fill('sin', math.sin(x))
        nt.fill('cos', math.cos(x))
        nt.write()
    return nt
Esempio n. 5
0
def test_access_row() :
    nt = NTuple('A','B')
    nt.fill('A', 5)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 5)
    nt.fill('B',6)
    ntl.assert_true(rw.A==5 and rw.B == 6)
Esempio n. 6
0
def reference_ntuple() :
    nt = NTuple('x', 'sin', 'cos')
    for x in xrange(-50,51) :
        nt.fill('x',x)
        nt.fill('sin', math.sin(x))
        nt.fill('cos', math.cos(x))
        nt.write()
    return nt
'''
Test suite for histogram.axis class.
'''

__author__ = "Juan PALACIOS [email protected]"

from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from pyhistuples.pyhistoplots import ntuple_plot, histo_plot, ntuple_column_histo

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

for x in xrange(10000):
    nt.fill('x', x)
    nt.fill('p', gauss(mu_p, sigma_p))
    nt.fill('pt', gauss(mu_pt, sigma_pt))
    nt.write()


def test_ntuple_plot():
    pt_plot = ntuple_plot(nt, 'pt', show=False)
    p_plot = ntuple_plot(nt,
                         'p',
                         errorfunction=lambda x: x.height / 2.,
                         color='red',
                         linewidth='1.5',
Esempio n. 8
0
import math
from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from matplotlib import pyplot
from pyhistuples.pyhistoplots import ntuple_plot, histo_plot, ntuple_column_histo
from pyhistuples.pyhistogram.histogram import Histogram, Axis

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

h_pt = Histogram(axis=Axis(100, -50, 50, label='pt'))

for x in xrange(10000):
    val = gauss(mu_pt, sigma_pt)
    h_pt.fill(val)

# greem plot without errors
plot_pt = histo_plot(h_pt, color='green', errorfunction=None)

# blue plot with default errors (poissonSigma)
plot_pt_errors = histo_plot(h_pt, color='blue')

print 'plot_pt range', plot_pt.axes[0].xaxis.get_view_interval()

print 'plot_pt_errors range', plot_pt_errors.axes[0].xaxis.get_view_interval()
Esempio n. 9
0
import math
from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from matplotlib import pyplot

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

for x in xrange(10000) :
    nt.fill('x',x)
    nt.fill('p', gauss(mu_p, sigma_p))
    nt.fill('pt', gauss(mu_pt, sigma_pt))
    nt.write()

x = nt.column('x', lambda row : row.p > 5. and row.pt > 1.)

Esempio n. 10
0
def test_access_latest_row() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 5)
    nt.fill('A', 7)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 7)
    nt.write()
    nt.fill('A', 22)
    rw = nt.row(1)
    ntl.assert_true(rw.A == 22)
Esempio n. 11
0
def test_fill_column_twice() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)
    nt.fill('A', 7)
Esempio n. 12
0
def test_illegal_tag_fails() :
    nt = NTuple('A','B')
    nt.fill('C', 5)
Esempio n. 13
0
def test_fill_column() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)
Esempio n. 14
0
def test_instantiate() :
    nt = NTuple('A', 'B')
    ntl.assert_true(nt is not None)
Esempio n. 15
0
def test_illegal_add_raises_KeyError() :
    nt0 = NTuple('A', 'B')
    nt1 = NTuple('C', 'D')
    _ = nt0 + nt1
import math
from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from matplotlib import pyplot
from pyhistuples.pyhistoplots import ntuple_plot, histo_plot, ntuple_column_histo

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

for x in xrange(10000) :
    nt.fill('x',x)
    nt.fill('p', gauss(mu_p, sigma_p))
    nt.fill('pt', gauss(mu_pt, sigma_pt))
    nt.write()


pt_plot = ntuple_plot(nt, 'pt')
h_pt = ntuple_column_histo(nt, 'pt')
histo_plot(h_pt, color='blue')