def test_VariantUnionArray(self): size = TestUtility.getRandomListSize() pv = PvObject({'v': [()]}) unionList = [] for i in range(0, size): value = TestUtility.getRandomInt() unionList.append(PvInt(value)) pv['v'] = unionList ul = pv['v'] for i in range(0, size): uli = ul[i][0] assert (uli['value'] == unionList[i]['value']) unionList.reverse() pv.setUnionArray(unionList) ul = pv.getUnionArray() for i in range(0, size): uli = ul[i] assert (uli['value'] == unionList[i]['value'])
def test_RestrictedUnionArray(self): size = TestUtility.getRandomListSize() pv = PvObject({'v': [({'i': INT, 's': STRING, 'd': DOUBLE}, )]}) unionList = [] for i in range(0, size): if i % 3 == 0: value = TestUtility.getRandomInt() unionList.append(PvObject({'i': INT}, {'i': value})) elif i % 3 == 1: value = TestUtility.getRandomString() unionList.append(PvObject({'s': STRING}, {'s': value})) else: value = TestUtility.getRandomDouble() unionList.append(PvObject({'d': DOUBLE}, {'d': value})) pv['v'] = unionList ul = pv['v'] for i in range(0, size): uli = ul[i][0] if 'i' in uli: assert (uli['i'] == unionList[i]['i']) elif 's' in uli: assert (uli['s'] == unionList[i]['s']) else: TestUtility.assertDoubleEquality(uli['d'], unionList[i]['d']) unionList.reverse() pv.setUnionArray(unionList) ul = pv['v'] for i in range(0, size): uli = ul[i][0] if 'i' in uli: assert (uli['i'] == unionList[i]['i']) elif 's' in uli: assert (uli['s'] == unionList[i]['s']) else: TestUtility.assertDoubleEquality(uli['d'], unionList[i]['d'])