test_rtcUFF Namespace Reference

Functions
def	relative_array_error (array1, array2)
def	test_initCentro_nvalid ()
def	test_initCentro_offset ()
def	test_initCentro_scale ()
def	test_initCentro_type ()
def	test_initControl_nslope ()
def	test_initControl_nactu ()
def	test_initControl_type ()
def	test_initControl_delay ()
def	test_set_npix ()
def	test_load_validposX ()
def	test_load_validposY ()
def	test_set_cmat ()
def	test_set_gain ()
def	test_load_img ()
def	test_set_dark ()
def	test_set_flat ()
def	test_calibrate_img ()
def	test_doCentroids_cog ()
def	test_do_control_generic ()
def	test_set_comRange ()
def	test_clipping ()
def	test_add_perturb_voltage ()
def	test_remove_perturb_voltage ()
def	test_add_perturb ()
def	test_disable_perturb_voltage ()
def	test_enable_perturb_voltage ()
def	test_reset_perturb_voltage ()
def	test_comp_voltage ()
def	test_remove_centroider ()
def	test_doCentroids_tcog ()
def	test_doCentroids_bpcog ()

Variables
int	precision = 1e-2
	config
	sup = Supervisor(config)
	rtc = Rtc()
	idx_centro
	ncentro
	centro = rtc.d_centro[0]
	control = rtc.d_control[0]
	xvalid = np.array(sup.rtc.rtc.d_centro[0].d_validx)
	yvalid = np.array(sup.rtc.rtc.d_centro[0].d_validy)
	cmat = sup.rtc.get_command_matrix(0)
	frame = np.array(sup.wfs.wfs.d_wfs[0].d_camimg)
	slp = ng.array(rtc.d_control[0].d_centroids)
	com = ng.array(rtc.d_control[0].d_com)
	dark = np.random.random(frame.shape)
	flat = np.random.random(frame.shape)

Function Documentation

relative_array_error()

def test_rtcUFF.relative_array_error	(		array1,
			array2
	)

Definition at line 54 of file test_rtcUFF.py.

def relative_array_error(array1, array2):
    return np.abs((array1 - array2) / array2.max()).max()
 
 

Here is the caller graph for this function:

test_add_perturb()

def test_rtcUFF.test_add_perturb

(

)

Definition at line 179 of file test_rtcUFF.py.

def test_add_perturb():
    C = np.random.random(sup.config.p_controller0.nactu)
    control.add_perturb_voltage("test", C, 1)
    com = ng.array(control.d_com_clipped).toarray()
    control.add_perturb()
    assert (relative_array_error(ng.array(control.d_com_clipped).toarray(), com + C) <
            precision)
 
 

Here is the call graph for this function:

test_add_perturb_voltage()

def test_rtcUFF.test_add_perturb_voltage

(

)

Definition at line 167 of file test_rtcUFF.py.

def test_add_perturb_voltage():
    C = np.random.random(sup.config.p_controller0.nactu)
    control.add_perturb_voltage("test", C, 1)
    assert (relative_array_error(
            ng.array(control.d_perturb_map["test"][0]).toarray(), C) < precision)
 
 

Here is the call graph for this function:

test_calibrate_img()

def test_rtcUFF.test_calibrate_img

(

)

Definition at line 122 of file test_rtcUFF.py.

def test_calibrate_img():
    centro.calibrate_img()
    imgCal = (frame - dark) * flat
    assert (relative_array_error(ng.array(centro.d_img).toarray(), imgCal) < precision)
 
 

Here is the call graph for this function:

test_clipping()

def test_rtcUFF.test_clipping

(

)

Definition at line 155 of file test_rtcUFF.py.

def test_clipping():
    control.set_comRange(-1, 1)
    C = (np.random.random(sup.config.p_controller0.nactu) - 0.5) * 4
    control.set_com(C, C.size)
    rtc.do_clipping(0)
    C_clipped = C.copy()
    C_clipped[np.where(C > 1)] = 1
    C_clipped[np.where(C < -1)] = -1
    assert (relative_array_error(ng.array(control.d_com_clipped).toarray(), C_clipped) <
            precision)
 
 

Here is the call graph for this function:

test_comp_voltage()

def test_rtcUFF.test_comp_voltage

(

)

Definition at line 209 of file test_rtcUFF.py.

def test_comp_voltage():
    volt_min = -1
    volt_max = 1
    control.set_comRange(volt_min, volt_max)
    control.comp_voltage()
    C = np.random.random(sup.config.p_controller0.nactu)
    control.add_perturb_voltage("test", C, 1)
    control.set_com(C, C.size)
    com0 = ng.array(control.d_circularComs0).toarray()
    com1 = ng.array(control.d_circularComs1).toarray()
    control.comp_voltage()
    delay = sup.config.p_controller0.delay
    a = delay - int(delay)
    b = 1 - a
    commands = a * com0 + b * com1
    comPertu = commands + C
    comPertu[np.where(comPertu > volt_max)] = volt_max
    comPertu[np.where(comPertu < volt_min)] = volt_min
    assert (relative_array_error(ng.array(control.d_voltage).toarray(), comPertu) <
            precision)
 
 

Here is the call graph for this function:

test_disable_perturb_voltage()

def test_rtcUFF.test_disable_perturb_voltage

(

)

Definition at line 188 of file test_rtcUFF.py.

def test_disable_perturb_voltage():
    control.disable_perturb_voltage("test")
    com = ng.array(control.d_com).toarray()
    control.add_perturb()
    assert (relative_array_error(ng.array(control.d_com).toarray(), com) < precision)
 
 

Here is the call graph for this function:

test_do_control_generic()

def test_rtcUFF.test_do_control_generic

(

)

Definition at line 141 of file test_rtcUFF.py.

def test_do_control_generic():
    slopes = ng.array(control.d_centroids).toarray()
    gain = control.gain
    cmat = ng.array(control.d_cmat).toarray()
    commands = cmat.dot(slopes) * gain * (-1)
    assert (relative_array_error(ng.array(control.d_com).toarray(), commands) <
            precision)
 
 

Here is the call graph for this function:

test_doCentroids_bpcog()

def test_rtcUFF.test_doCentroids_bpcog

(

)

Definition at line 266 of file test_rtcUFF.py.

def test_doCentroids_bpcog():
    rtc.remove_centroider(0)
    rtc.add_centroider(sup.context, sup.config.p_wfs0._nvalid,
                       sup.config.p_wfs0.npix / 2 - 0.5, sup.config.p_wfs0.pixsize,
                       False, 0, "bpcog")
 
    centro = rtc.d_centro[-1]
    bpix = 8
    centro.set_nmax(8)
    centro.set_npix(sup.config.p_wfs0.npix)
    centro.load_validpos(xvalid, yvalid, xvalid.size)
    centro.load_img(frame, frame.shape[0])
    centro.calibrate_img()
    rtc.do_centroids(0)
    bincube = np.array(sup.wfs.wfs.d_wfs[0].d_bincube)
    bincube /= bincube.max()
    slopes = np.zeros(sup.config.p_wfs0._nvalid * 2)
    offset = centro.offset
    scale = centro.scale
    vx = sup.config.p_wfs0._validsubsx
    vy = sup.config.p_wfs0._validsubsy
    npix = sup.config.p_wfs0.npix
    for k in range(sup.config.p_wfs0._nvalid):
        imagette = frame[vx[k]:vx[k] + npix, vy[k]:vy[k] + npix].astype(np.float32)
        threshold = np.sort(imagette, axis=None)[-(bpix + 1)]
        imagette -= threshold
        imagette[np.where(imagette < 0)] = 0
        tmp = center_of_mass(imagette)
        slopes[k] = (tmp[0] - offset) * scale
        slopes[k + sup.config.p_wfs0._nvalid] = (tmp[1] - offset) * scale
    assert (relative_array_error(ng.array(control.d_centroids).toarray(), slopes) <
            precision)

Here is the call graph for this function:

test_doCentroids_cog()

def test_rtcUFF.test_doCentroids_cog

(

)

Definition at line 128 of file test_rtcUFF.py.

def test_doCentroids_cog():
    bincube = np.array(sup.wfs.wfs.d_wfs[0].d_bincube)
    slopes = np.zeros(sup.config.p_wfs0._nvalid * 2)
    offset = centro.offset
    scale = centro.scale
    for k in range(sup.config.p_wfs0._nvalid):
        tmp = center_of_mass(bincube[:, :, k])
        slopes[k] = (tmp[0] - offset) * scale
        slopes[k + sup.config.p_wfs0._nvalid] = (tmp[1] - offset) * scale
    assert (relative_array_error(ng.array(control.d_centroids).toarray(), slopes) <
            precision)
 
 

Here is the call graph for this function:

test_doCentroids_tcog()

def test_rtcUFF.test_doCentroids_tcog

(

)

Definition at line 236 of file test_rtcUFF.py.

def test_doCentroids_tcog():
    rtc.add_centroider(sup.context, sup.config.p_wfs0._nvalid,
                       sup.config.p_wfs0.npix / 2 - 0.5, sup.config.p_wfs0.pixsize,
                       False, 0, "tcog")
 
    centro = rtc.d_centro[-1]
    threshold = 5000
    centro.set_threshold(threshold)
    centro.set_npix(sup.config.p_wfs0.npix)
    centro.load_validpos(xvalid, yvalid, xvalid.size)
    centro.load_img(frame, frame.shape[0])
    centro.calibrate_img()
    rtc.do_centroids(0)
    slopes = np.zeros(sup.config.p_wfs0._nvalid * 2)
    offset = centro.offset
    scale = centro.scale
    vx = sup.config.p_wfs0._validsubsx
    vy = sup.config.p_wfs0._validsubsy
    npix = sup.config.p_wfs0.npix
    for k in range(sup.config.p_wfs0._nvalid):
        imagette = frame[vx[k]:vx[k] + npix, vy[k]:vy[k] + npix].astype(
                np.float32) - threshold
        imagette[np.where(imagette < 0)] = 0
        tmp = center_of_mass(imagette)
        slopes[k] = (tmp[0] - offset) * scale
        slopes[k + sup.config.p_wfs0._nvalid] = (tmp[1] - offset) * scale
    assert (relative_array_error(ng.array(control.d_centroids).toarray(), slopes) <
            precision)
 
 

Here is the call graph for this function:

test_enable_perturb_voltage()

def test_rtcUFF.test_enable_perturb_voltage

(

)

Definition at line 195 of file test_rtcUFF.py.

def test_enable_perturb_voltage():
    control.enable_perturb_voltage("test")
    com = ng.array(control.d_com_clipped).toarray()
    C = ng.array(control.d_perturb_map["test"][0]).toarray()
    control.add_perturb()
    assert (relative_array_error(ng.array(control.d_com_clipped).toarray(), com + C) <
            precision)
 
 

Here is the call graph for this function:

test_initCentro_nvalid()

def test_rtcUFF.test_initCentro_nvalid

(

)

Definition at line 58 of file test_rtcUFF.py.

def test_initCentro_nvalid():
    assert (centro.nvalid - sup.config.p_wfs0._nvalid < precision)
 
 

test_initCentro_offset()

def test_rtcUFF.test_initCentro_offset

(

)

Definition at line 62 of file test_rtcUFF.py.

def test_initCentro_offset():
    assert (centro.offset - (sup.config.p_wfs0.npix / 2 - 0.5) < precision)
 
 

test_initCentro_scale()

def test_rtcUFF.test_initCentro_scale

(

)

Definition at line 66 of file test_rtcUFF.py.

def test_initCentro_scale():
    assert (centro.scale - sup.config.p_wfs0.pixsize < precision)
 
 

test_initCentro_type()

def test_rtcUFF.test_initCentro_type

(

)

Definition at line 70 of file test_rtcUFF.py.

def test_initCentro_type():
    assert (centro.type == "cog")
 
 

test_initControl_delay()

def test_rtcUFF.test_initControl_delay

(

)

Definition at line 86 of file test_rtcUFF.py.

def test_initControl_delay():
    assert (control.delay - sup.config.p_controller0.delay < precision)
 
 

test_initControl_nactu()

def test_rtcUFF.test_initControl_nactu

(

)

Definition at line 78 of file test_rtcUFF.py.

def test_initControl_nactu():
    assert (control.nactu - sup.config.p_controller0.nactu < precision)
 
 

test_initControl_nslope()

def test_rtcUFF.test_initControl_nslope

(

)

Definition at line 74 of file test_rtcUFF.py.

def test_initControl_nslope():
    assert (control.nslope - sup.config.p_wfs0._nvalid * 2 < precision)
 
 

test_initControl_type()

def test_rtcUFF.test_initControl_type

(

)

Definition at line 82 of file test_rtcUFF.py.

def test_initControl_type():
    assert (control.type == "generic")
 
 

test_load_img()

def test_rtcUFF.test_load_img

(

)

Definition at line 110 of file test_rtcUFF.py.

def test_load_img():
    assert (relative_array_error(np.array(centro.d_img_raw), frame) < precision)
 
 

Here is the call graph for this function:

test_load_validposX()

def test_rtcUFF.test_load_validposX

(

)

Definition at line 94 of file test_rtcUFF.py.

def test_load_validposX():
    assert (relative_array_error(np.array(centro.d_validx), xvalid) < precision)
 
 

Here is the call graph for this function:

test_load_validposY()

def test_rtcUFF.test_load_validposY

(

)

Definition at line 98 of file test_rtcUFF.py.

def test_load_validposY():
    assert (relative_array_error(np.array(centro.d_validy), yvalid) < precision)
 
 

Here is the call graph for this function:

test_remove_centroider()

def test_rtcUFF.test_remove_centroider

(

)

Definition at line 231 of file test_rtcUFF.py.

def test_remove_centroider():
    rtc.remove_centroider(0)
    assert (rtc.d_centro == [])
 
 

test_remove_perturb_voltage()

def test_rtcUFF.test_remove_perturb_voltage

(

)

Definition at line 174 of file test_rtcUFF.py.

def test_remove_perturb_voltage():
    control.remove_perturb_voltage("test")
    assert (control.d_perturb_map == {})
 
 

test_reset_perturb_voltage()

def test_rtcUFF.test_reset_perturb_voltage

(

)

Definition at line 204 of file test_rtcUFF.py.

def test_reset_perturb_voltage():
    control.reset_perturb_voltage()
    assert (control.d_perturb_map == {})
 
 

test_set_cmat()

def test_rtcUFF.test_set_cmat

(

)

Definition at line 102 of file test_rtcUFF.py.

def test_set_cmat():
    assert (relative_array_error(ng.array(control.d_cmat).toarray(), cmat) < precision)
 
 

Here is the call graph for this function:

test_set_comRange()

def test_rtcUFF.test_set_comRange

(

)

Definition at line 150 of file test_rtcUFF.py.

def test_set_comRange():
    control.set_comRange(-1, 1)
    assert (control.comRange == (-1, 1))
 
 

test_set_dark()

def test_rtcUFF.test_set_dark

(

)

Definition at line 114 of file test_rtcUFF.py.

def test_set_dark():
    assert (relative_array_error(ng.array(centro.d_dark).toarray(), dark) < precision)
 
 

Here is the call graph for this function:

test_set_flat()

def test_rtcUFF.test_set_flat

(

)

Definition at line 118 of file test_rtcUFF.py.

def test_set_flat():
    assert (relative_array_error(ng.array(centro.d_flat).toarray(), flat) < precision)
 
 

Here is the call graph for this function:

test_set_gain()

def test_rtcUFF.test_set_gain

(

)

Definition at line 106 of file test_rtcUFF.py.

def test_set_gain():
    assert (control.gain - sup.config.p_controller0.gain < precision)
 
 

test_set_npix()

def test_rtcUFF.test_set_npix

(

)

Definition at line 90 of file test_rtcUFF.py.

def test_set_npix():
    assert (centro.npix - sup.config.p_wfs0.npix < precision)
 
 

Variable Documentation

centro

test_rtcUFF.centro = rtc.d_centro[0]

Definition at line 30 of file test_rtcUFF.py.

cmat

test_rtcUFF.cmat = sup.rtc.get_command_matrix(0)

Definition at line 36 of file test_rtcUFF.py.

com

test_rtcUFF.com = ng.array(rtc.d_control[0].d_com)

Definition at line 46 of file test_rtcUFF.py.

config

test_rtcUFF.config

Initial value:

=  load_config_from_file(os.getenv("COMPASS_ROOT") +
        "/shesha/tests/pytest/par/test_sh.py")

Definition at line 11 of file test_rtcUFF.py.

control

test_rtcUFF.control = rtc.d_control[0]

Definition at line 31 of file test_rtcUFF.py.

dark

test_rtcUFF.dark = np.random.random(frame.shape)

Definition at line 48 of file test_rtcUFF.py.

flat

test_rtcUFF.flat = np.random.random(frame.shape)

Definition at line 49 of file test_rtcUFF.py.

frame

test_rtcUFF.frame = np.array(sup.wfs.wfs.d_wfs[0].d_camimg)

Definition at line 39 of file test_rtcUFF.py.

idx_centro

test_rtcUFF.idx_centro

Definition at line 28 of file test_rtcUFF.py.

ncentro

test_rtcUFF.ncentro

Definition at line 29 of file test_rtcUFF.py.

precision

int test_rtcUFF.precision = 1e-2

Definition at line 9 of file test_rtcUFF.py.

rtc

test_rtcUFF.rtc = Rtc()

Definition at line 22 of file test_rtcUFF.py.

slp

test_rtcUFF.slp = ng.array(rtc.d_control[0].d_centroids)

Definition at line 44 of file test_rtcUFF.py.

sup

test_rtcUFF.sup = Supervisor(config)

Definition at line 13 of file test_rtcUFF.py.

xvalid

test_rtcUFF.xvalid = np.array(sup.rtc.rtc.d_centro[0].d_validx)

Definition at line 33 of file test_rtcUFF.py.

yvalid

test_rtcUFF.yvalid = np.array(sup.rtc.rtc.d_centro[0].d_validy)

Definition at line 34 of file test_rtcUFF.py.