def test_waveform_get_with_count_arg(self):
with no_simulator_updates():
wf = PV(pvnames.char_arr_pv, count=32)
val = wf.get()
self.assertEquals(len(val), 32)
val = wf.get(count=wf.nelm)
self.assertEquals(len(val), wf.nelm)
def test_get1(self):
write('Simple Test: test value and char_value on an integer\n')
with no_simulator_updates():
pv = PV(pvnames.int_pv)
val = pv.get()
cval = pv.get(as_string=True)
self.failUnless(int(cval) == val)
def testA_CreatedWithConn(self):
write('Simple Test: create pv with conn callback\n')
pv = PV(pvnames.int_pv, connection_callback=onConnect)
val = pv.get()
global CONN_DAT
conn = CONN_DAT.get(pvnames.int_pv, None)
self.assertEqual(conn, True)
def test_put_string_waveform(self):
write('String Array: \n')
with no_simulator_updates():
pv = PV(pvnames.string_arr_pv)
put_value = ['a', 'b', 'c']
pv.put(put_value)
get_value = pv.get(use_monitor=False, count=len(put_value))
numpy.testing.assert_array_equal(get_value, put_value)
def test_get_string_waveform(self):
write('String Array: \n')
with no_simulator_updates():
pv = PV(pvnames.string_arr_pv)
val = pv.get()
self.failUnless(len(val) > 10)
self.assertIsInstance(val[0], str)
self.failUnless(len(val[0]) > 1)
self.assertIsInstance(val[1], str)
self.failUnless(len(val[1]) > 1)
def test_DoubleVal(self):
pvn = pvnames.double_pv
pv = PV(pvn)
pv.get()
cdict = pv.get_ctrlvars()
write('Testing CTRL Values for a Double (%s)\n' % (pvn))
self.failUnless('severity' in cdict)
self.failUnless(len(pv.host) > 1)
self.assertEqual(pv.count, 1)
self.assertEqual(pv.precision, pvnames.double_pv_prec)
self.assertEqual(pv.units, pvnames.double_pv_units)
self.failUnless(pv.access.startswith('read'))
def threading():
nonlocal pvs
pvs = [
PV(pvname, auto_monitor=False, context=context)
for pvname in pv_names
]
while not all(pv.connected for pv in pvs):
time.sleep(0.01)
def test_putcomplete(self):
write('Put with wait and put_complete (using real motor!) \n')
vals = (1.35, 1.50, 1.44, 1.445, 1.45, 1.453, 1.446, 1.447, 1.450, 1.450, 1.490, 1.5, 1.500)
p = PV(pvnames.motor1)
# this works with a real motor, fail if it doesn't connect quickly
if not p.wait_for_connection(timeout=0.2):
self.skipTest('Unable to connect to real motor record')
see_complete = []
for v in vals:
t0 = time.time()
p.put(v, use_complete=True)
count = 0
for i in range(100000):
time.sleep(0.001)
count = count + 1
if p.put_complete:
see_complete.append(True)
break
# print( 'made it to value= %.3f, elapsed time= %.4f sec (count=%i)' % (v, time.time()-t0, count))
self.failUnless(len(see_complete) > (len(vals) - 5))
def bench_threading_get_speed(pvname,
*,
initial_value=None,
log_level='ERROR'):
from caproto.threading.client import (PV, SharedBroadcaster, Context as
ThreadingContext)
shared_broadcaster = SharedBroadcaster()
context = ThreadingContext(broadcaster=shared_broadcaster,
log_level=log_level)
def threading():
value = pv.get(use_monitor=False)
if initial_value is not None:
assert len(value) == len(initial_value)
pv = PV(pvname, auto_monitor=False, context=context)
if initial_value is not None:
pv.put(initial_value, wait=True)
yield threading
logger.debug('Disconnecting threading pv %s', pv)
pv.disconnect()
logger.debug('Disconnecting shared broadcaster %s', shared_broadcaster)
shared_broadcaster.disconnect()
logger.debug('Done')
def bench_threading_put_speed(pvname, *, value, log_level='ERROR'):
from caproto.threading.client import (PV, SharedBroadcaster, Context as
ThreadingContext)
shared_broadcaster = SharedBroadcaster()
context = ThreadingContext(broadcaster=shared_broadcaster,
log_level=log_level)
def threading():
pv.put(value, wait=True)
pv = PV(pvname, auto_monitor=False, context=context)
yield threading
np.testing.assert_array_almost_equal(pv.get(), value)
logger.debug('Disconnecting threading pv %s', pv)
pv.disconnect()
logger.debug('Disconnecting shared broadcaster %s', shared_broadcaster)
shared_broadcaster.disconnect()
logger.debug('Done')
def testEnumPut(self):
pv = PV(pvnames.enum_pv)
self.assertIsNot(pv, None)
pv.put('Stop')
time.sleep(0.1)
val = pv.get()
self.assertEqual(val, 0)
assert pv.get(as_string=True) == 'Stop'
def test_subarray_zerolen(self):
subarr1 = PV(pvnames.zero_len_subarr1)
subarr1.wait_for_connection()
val = subarr1.get(use_monitor=True, as_numpy=True)
self.assertIsInstance(val, numpy.ndarray, msg='using monitor')
self.assertEquals(len(val), 0, msg='using monitor')
# caproto returns things in big endian, not native type
# self.assertEquals(val.dtype, numpy.float64, msg='using monitor')
val = subarr1.get(use_monitor=False, as_numpy=True)
self.assertIsInstance(val, numpy.ndarray, msg='no monitor')
self.assertEquals(len(val), 0, msg='no monitor')
def test_subarray_1elem(self):
with no_simulator_updates():
# pv = PV(pvnames.zero_len_subarr1)
pv = PV(pvnames.double_arr_pv)
pv.wait_for_connection()
val = pv.get(count=1, use_monitor=False)
print('val is', val, type(val))
self.assertIsInstance(val, numpy.ndarray)
self.assertEqual(len(val), 1)
val = pv.get(count=1, as_numpy=False, use_monitor=False)
print('val is', val, type(val))
self.assertIsInstance(val, list)
self.assertEqual(len(val), 1)
def test_get_callback(self):
write("Callback test: changing PV must be updated\n")
global NEWVALS
mypv = PV(pvnames.updating_pv1)
NEWVALS = []
def onChanges(pvname=None, value=None, char_value=None, **kw):
write('PV %s %s, %s Changed!\n' % (pvname, repr(value), char_value))
NEWVALS.append(repr(value))
mypv.add_callback(onChanges)
write('Added a callback. Now wait for changes...\n')
t0 = time.time()
while time.time() - t0 < 3:
time.sleep(1.e-4)
write(' saw %i changes.\n' % len(NEWVALS))
assert len(NEWVALS) > 3
mypv.clear_callbacks()
def test_waveform_callback_with_count_arg(self):
values = []
wf = PV(pvnames.char_arr_pv, count=32)
def onChanges(pvname=None, value=None, char_value=None, **kw):
write( 'PV %s %s, %s Changed!\n' % (pvname, repr(value), char_value))
values.append( value)
wf.add_callback(onChanges)
write('Added a callback. Now wait for changes...\n')
t0 = time.time()
while time.time() - t0 < 3:
time.sleep(1.e-4)
if len(values)>0:
break
self.failUnless(len(values) > 0)
self.assertEquals(len(values[0]),32)
wf.clear_callbacks()
def test_waveform_get_1elem(self):
pv = PV(pvnames.double_arr_pv)
val = pv.get(count=1, use_monitor=False)
self.failUnless(isinstance(val, numpy.ndarray))
self.failUnless(len(val), 1)
def test_emptyish_char_waveform_monitor(self):
'''a test of a char waveform of length 1 (NORD=1): value "\0"
with using auto_monitor
'''
with no_simulator_updates():
zerostr = PV(pvnames.char_arr_pv, auto_monitor=True)
zerostr.wait_for_connection()
zerostr.put([0], wait=True)
time.sleep(0.2)
self.assertEquals(zerostr.get(as_string=True), '')
numpy.testing.assert_array_equal(zerostr.get(as_string=False), [0])
self.assertEquals(zerostr.get(as_string=True, as_numpy=False), '')
numpy.testing.assert_array_equal(
zerostr.get(as_string=False, as_numpy=False), [0])
zerostr.put([0, 0], wait=True)
time.sleep(0.2)
self.assertEquals(zerostr.get(as_string=True), '')
numpy.testing.assert_array_equal(zerostr.get(as_string=False),
[0, 0])
self.assertEquals(zerostr.get(as_string=True, as_numpy=False), '')
numpy.testing.assert_array_equal(
zerostr.get(as_string=False, as_numpy=False), [0, 0])
zerostr.disconnect()
def test_emptyish_char_waveform_no_monitor(self):
'''a test of a char waveform of length 1 (NORD=1): value "\0"
without using auto_monitor
'''
with no_simulator_updates():
zerostr = PV(pvnames.char_arr_pv, auto_monitor=False)
zerostr.wait_for_connection()
# elem_count = 128, requested count = None, libca returns count = 1
zerostr.put([0], wait=True)
self.assertEquals(zerostr.get(as_string=True), '')
numpy.testing.assert_array_equal(zerostr.get(as_string=False), [0])
self.assertEquals(zerostr.get(as_string=True, as_numpy=False), '')
numpy.testing.assert_array_equal(
zerostr.get(as_string=False, as_numpy=False), [0])
# elem_count = 128, requested count = None, libca returns count = 2
zerostr.put([0, 0], wait=True)
self.assertEquals(zerostr.get(as_string=True), '')
numpy.testing.assert_array_equal(zerostr.get(as_string=False),
[0, 0])
self.assertEquals(zerostr.get(as_string=True, as_numpy=False), '')
numpy.testing.assert_array_equal(
zerostr.get(as_string=False, as_numpy=False), [0, 0])
def test_subarrays(self):
write("Subarray test: dynamic length arrays\n")
driver = PV(pvnames.subarr_driver)
subarr1 = PV(pvnames.subarr1)
subarr1.connect()
len_full = 64
len_sub1 = 16
full_data = numpy.arange(len_full) / 1.0
caput("%s.NELM" % pvnames.subarr1, len_sub1)
caput("%s.INDX" % pvnames.subarr1, 0)
driver.put(full_data)
time.sleep(0.1)
subval = subarr1.get()
self.assertEqual(len(subval), len_sub1)
self.failUnless(numpy.all(subval == full_data[:len_sub1]))
write("Subarray test: C\n")
caput("%s.NELM" % pvnames.subarr2, 19)
caput("%s.INDX" % pvnames.subarr2, 3)
subarr2 = PV(pvnames.subarr2)
subarr2.get()
driver.put(full_data)
time.sleep(0.1)
subval = subarr2.get()
self.assertEqual(len(subval), 19)
self.failUnless(numpy.all(subval == full_data[3:3 + 19]))
caput("%s.NELM" % pvnames.subarr2, 5)
caput("%s.INDX" % pvnames.subarr2, 13)
driver.put(full_data)
time.sleep(0.1)
subval = subarr2.get()
self.assertEqual(len(subval), 5)
self.failUnless(numpy.all(subval == full_data[13:5 + 13]))
def testA_CreatePV(self):
write('Simple Test: create pv\n')
pv = PV(pvnames.double_pv)
self.assertIsNot(pv, None)
def test_thread_pv(threading_broadcaster):
from caproto.threading.client import Context, PV
pv1 = "XF:31IDA-OP{Tbl-Ax:X1}Mtr.VAL"
# pv2 = "XF:31IDA-OP{Tbl-Ax:X2}Mtr.VAL"
# Some user function to call when subscriptions receive data.
called = []
def user_callback(*, value, **kwargs):
print()
print('-- user callback', value)
called.append(True)
ctx = Context(threading_broadcaster, log_level='DEBUG')
ctx.register()
time_pv = PV(pv1, context=ctx, form='time')
ctrl_pv = PV(pv1, context=ctx, form='ctrl')
time_pv.wait_for_connection()
time_pv.add_callback(user_callback)
print('time read', time_pv.get())
print('ctrl read', ctrl_pv.get())
time_pv.put(3, wait=True)
time_pv.put(6, wait=True)
time.sleep(0.1)
assert time_pv.get() == 6
assert called
print('read', time_pv.get())
print('done')
repr(time_pv)
for k, v in PV.__dict__.items():
if isinstance(v, property):
getattr(time_pv, k)
getattr(ctrl_pv, k)
def test_putwait(self):
write('Put with wait (using real motor!) \n')
pv = PV(pvnames.motor1)
# this works with a real motor, fail if it doesn't connect quickly
if not pv.wait_for_connection(timeout=0.2):
self.skipTest('Unable to connect to real motor record')
val = pv.get()
t0 = time.time()
if val < 5:
pv.put(val + 1.0, wait=True)
else:
pv.put(val - 1.0, wait=True)
dt = time.time() - t0
write(' put took %s sec\n' % dt)
self.failUnless(dt > 0.1)
# now with a callback!
global put_callback_called
put_callback_called = False
def onPutdone(pvname=None, **kws):
print('put done ', pvname, kws)
global put_callback_called
put_callback_called = True
val = pv.get()
if val < 5:
pv.put(val + 1.0, callback=onPutdone)
else:
pv.put(val - 1.0, callback=onPutdone)
t0 = time.time()
while time.time() - t0 < dt * 1.50:
time.sleep(0.02)
write(' put should be done by now? %s \n' % put_callback_called)
self.failUnless(put_callback_called)
# now using pv.put_complete
val = pv.get()
if val < 5:
pv.put(val + 1.0, use_complete=True)
else:
pv.put(val - 1.0, use_complete=True)
t0 = time.time()
count = 0
while time.time() - t0 < dt * 1.50:
if pv.put_complete:
break
count = count + 1
time.sleep(0.02)
write(
' put_complete=%s (should be True), and count=%i (should be>3)\n'
% (pv.put_complete, count))
self.failUnless(pv.put_complete)
self.failUnless(count > 3)
