def test_arraysqueeze_GoodArrayArgWrap(self):
import numpy as np
from opan.utils import pack_tups
# Store the array result for detailed checking
ary = np.array(range(1, 9, 2)).reshape((4, 1))
ret = self.fxn_2p_1d(ary, "x")
# Confirm converted array matches expectations
for i, t in enumerate(pack_tups(ary, ret[0])):
self.assertEqual(
t[0][0],
t[1],
msg="Mismatch at position {0} (Pre: {1}; Post: {2})".format(
i, t[0][0], t[1]))
def test_Utils_PackTupsTestAllSingletons(self):
from opan.utils import pack_tups
tups = pack_tups(0, 1, 2, 3, 4)
self.assertEqual(len(tups), 1)
self.assertTupleEqual(tups[0], tuple(range(5)))
def test_Utils_PackTupsStrNoIter(self):
from opan.utils import pack_tups
tups = pack_tups("ab", range(2))
self.assertEqual(tups[0][0], "ab")
self.assertEqual(tups[1][0], "ab")
def test_Utils_PackTupsGoodPacking(self):
from opan.utils import pack_tups
tups = pack_tups(range(3), range(3, 6), range(6, 9))
[[self.assertEqual(tups[i][j], 3 * j + i) for i in range(3)]
for j in range(3)]
def test_Utils_PackTupsTestAllSingletons(self):
from opan.utils import pack_tups
tups = pack_tups(0,1,2,3,4)
self.assertEqual(len(tups), 1)
self.assertTupleEqual(tups[0], tuple(range(5)))
def test_Utils_PackTupsStrNoIter(self):
from opan.utils import pack_tups
tups = pack_tups("ab", range(2))
self.assertEqual(tups[0][0], "ab")
self.assertEqual(tups[1][0], "ab")
def test_Utils_PackTupsGoodPacking(self):
from opan.utils import pack_tups
tups = pack_tups(range(3), range(3,6), range(6,9))
[[self.assertEqual(tups[i][j], 3*j + i) for i in range(3)]
for j in range(3)]
