/usr/share/pyshared/neo/test/test_eventarray.py is in python-neo 0.3.3-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | # -*- coding: utf-8 -*-
"""
Tests of the neo.core.eventarray.EventArray class
"""
try:
import unittest2 as unittest
except ImportError:
import unittest
import numpy as np
import quantities as pq
from neo.core.eventarray import EventArray
from neo.test.tools import (assert_neo_object_is_compliant,
assert_arrays_equal, assert_same_sub_schema)
class TestEventArray(unittest.TestCase):
def test_EventArray_creation(self):
params = {'testarg2': 'yes', 'testarg3': True}
evta = EventArray([1.1, 1.5, 1.7]*pq.ms,
labels=np.array(['test event 1',
'test event 2',
'test event 3'], dtype='S'),
name='test', description='tester',
file_origin='test.file',
testarg1=1, **params)
evta.annotate(testarg1=1.1, testarg0=[1, 2, 3])
assert_neo_object_is_compliant(evta)
assert_arrays_equal(evta.times, [1.1, 1.5, 1.7]*pq.ms)
assert_arrays_equal(evta.labels, np.array(['test event 1',
'test event 2',
'test event 3'], dtype='S'))
self.assertEqual(evta.name, 'test')
self.assertEqual(evta.description, 'tester')
self.assertEqual(evta.file_origin, 'test.file')
self.assertEqual(evta.annotations['testarg0'], [1, 2, 3])
self.assertEqual(evta.annotations['testarg1'], 1.1)
self.assertEqual(evta.annotations['testarg2'], 'yes')
self.assertTrue(evta.annotations['testarg3'])
def test_EventArray_repr(self):
params = {'testarg2': 'yes', 'testarg3': True}
evta = EventArray([1.1, 1.5, 1.7]*pq.ms,
labels=np.array(['test event 1',
'test event 2',
'test event 3'], dtype='S'),
name='test', description='tester',
file_origin='test.file',
testarg1=1, **params)
evta.annotate(testarg1=1.1, testarg0=[1, 2, 3])
assert_neo_object_is_compliant(evta)
targ = ('<EventArray: test event 1@1.1 ms, test event 2@1.5 ms, ' +
'test event 3@1.7 ms>')
res = repr(evta)
self.assertEqual(targ, res)
def test_EventArray_merge(self):
params1 = {'testarg2': 'yes', 'testarg3': True}
params2 = {'testarg2': 'no', 'testarg4': False}
paramstarg = {'testarg2': 'yes;no',
'testarg3': True,
'testarg4': False}
epca1 = EventArray([1.1, 1.5, 1.7]*pq.ms,
labels=np.array(['test event 1 1',
'test event 1 2',
'test event 1 3'], dtype='S'),
name='test', description='tester 1',
file_origin='test.file',
testarg1=1, **params1)
epca2 = EventArray([2.1, 2.5, 2.7]*pq.us,
labels=np.array(['test event 2 1',
'test event 2 2',
'test event 2 3'], dtype='S'),
name='test', description='tester 2',
file_origin='test.file',
testarg1=1, **params2)
epcatarg = EventArray([1.1, 1.5, 1.7, .0021, .0025, .0027]*pq.ms,
labels=np.array(['test event 1 1',
'test event 1 2',
'test event 1 3',
'test event 2 1',
'test event 2 2',
'test event 2 3'], dtype='S'),
name='test',
description='merge(tester 1, tester 2)',
file_origin='test.file',
testarg1=1, **paramstarg)
assert_neo_object_is_compliant(epca1)
assert_neo_object_is_compliant(epca2)
assert_neo_object_is_compliant(epcatarg)
epcares = epca1.merge(epca2)
assert_neo_object_is_compliant(epcares)
assert_same_sub_schema(epcatarg, epcares)
if __name__ == "__main__":
unittest.main()
|