def test_basis_element_labels(self):
basisnames = ['gm', 'std', 'pp']
# One dimensional gm
self.assertEqual([''], bt.basis_element_labels('gm', 1))
# Two dimensional
expectedLabels = [
['I', 'X', 'Y', 'Z'],
['(0,0)', '(0,1)', '(1,0)', '(1,1)'],
['I', 'X', 'Y', 'Z']]
labels = [bt.basis_element_labels(basisname, 2) for basisname in basisnames]
self.assertEqual(labels, expectedLabels)
with self.assertRaises(NotImplementedError):
bt.basis_element_labels('asdklfasdf', 2)
# Non power of two for pp labels:
with self.assertRaises(ValueError):
label = bt.basis_element_labels('pp', 3)
# Single list arg for pp labels
self.assertEqual(bt.basis_element_labels('pp', [2]), ['I', 'X', 'Y', 'Z'])
# Four dimensional+
expectedLabels = [['I^{(0)}', 'X^{(0)}_{0,1}', 'X^{(0)}_{0,2}', 'X^{(0)}_{0,3}', 'X^{(0)}_{1,2}', 'X^{(0)}_{1,3}', 'X^{(0)}_{2,3}', 'Y^{(0)}_{0,1}', 'Y^{(0)}_{0,2}', 'Y^{(0)}_{0,3}', 'Y^{(0)}_{1,2}', 'Y^{(0)}_{1,3}', 'Y^{(0)}_{2,3}', 'Z^{(0)}_{1}', 'Z^{(0)}_{2}', 'Z^{(0)}_{3}'], ['(0,0)', '(0,1)', '(0,2)', '(0,3)', '(1,0)', '(1,1)', '(1,2)', '(1,3)', '(2,0)', '(2,1)', '(2,2)', '(2,3)', '(3,0)', '(3,1)', '(3,2)', '(3,3)'], ['II', 'IX', 'IY', 'IZ', 'XI', 'XX', 'XY', 'XZ', 'YI', 'YX', 'YY', 'YZ', 'ZI', 'ZX', 'ZY', 'ZZ']]
labels = [bt.basis_element_labels(basisname, 4) for basisname in basisnames]
self.assertEqual(expectedLabels, labels)
def test_basis_object(self):
#test a few aspects of a Basis object that other tests miss...
b = Basis("pp",2)
beq = b.expanded_equivalent()
longnm = bt.basis_longname(b)
lbls = bt.basis_element_labels(b)
raw_mxs = bt.basis_matrices("pp",2)
with self.assertRaises(NotImplementedError):
bt.basis_matrices("foobar",2) #invalid basis name
print("Dim = ", repr(b.dim) ) # calls Dim.__repr__
def test_basis_element_labels(self):
basisnames = ['gm', 'std', 'pp', 'akdlfjalsdf']
# One dimensional gm
self.assertEqual([''], basistools.basis_element_labels('gm', 1))
# Two dimensional
expectedLabels = [
['I', 'X', 'Y', 'Z'],
['(0,0)', '(0,1)', '(1,0)', '(1,1)'],
['I', 'X', 'Y', 'Z'],
[]]
labels = [basistools.basis_element_labels(basisname, 2) for basisname in basisnames]
self.assertEqual(labels, expectedLabels)
# Non power of two for pp labels:
with self.assertRaises(ValueError):
label = basistools.basis_element_labels('pp', 3)
with self.assertRaises(ValueError):
label = basistools.basis_element_labels('pp', [1, 2])
# Single list arg for pp labels
self.assertEqual(basistools.basis_element_labels('pp', [2]), ['I', 'X', 'Y', 'Z'])
# Four dimensional+
expectedLabels = [['I^{(0)}', 'X^{(0)}_{0,1}', 'X^{(0)}_{0,2}', 'X^{(0)}_{0,3}', 'X^{(0)}_{1,2}', 'X^{(0)}_{1,3}', 'X^{(0)}_{2,3}', 'Y^{(0)}_{0,1}', 'Y^{(0)}_{0,2}', 'Y^{(0)}_{0,3}', 'Y^{(0)}_{1,2}', 'Y^{(0)}_{1,3}', 'Y^{(0)}_{2,3}', 'Z^{(0)}_{1}', 'Z^{(0)}_{2}', 'Z^{(0)}_{3}'], ['(0,0)', '(0,1)', '(0,2)', '(0,3)', '(1,0)', '(1,1)', '(1,2)', '(1,3)', '(2,0)', '(2,1)', '(2,2)', '(2,3)', '(3,0)', '(3,1)', '(3,2)', '(3,3)'], ['II', 'IX', 'IY', 'IZ', 'XI', 'XX', 'XY', 'XZ', 'YI', 'YX', 'YY', 'YZ', 'ZI', 'ZX', 'ZY', 'ZZ'], []]
labels = [basistools.basis_element_labels(basisname, 4) for basisname in basisnames]
self.assertEqual(expectedLabels, labels)
def test_basis_object(self):
# test a few aspects of a Basis object that other tests miss...
b = basis.Basis.cast("pp", 4)
beq = b.simple_equivalent()
longnm = bt.basis_longname(b)
lbls = bt.basis_element_labels(b, None)
raw_mxs = bt.basis_matrices("pp", 4)
# TODO assert correctness for all
with self.assertRaises(AssertionError):
bt.basis_matrices("foobar", 4) # invalid basis name
print("Dim = ", repr(b.dim)) # calls Dim.__repr__