def test_packArrays(self): """JointEnumeration packArrays should return correct array.""" a = JointEnumeration(['xyz', 'abcd', 'ef']) v = [[0,1,2,0],[3,3,1,0], [1,1,0,0]] result = a.packArrays(v) self.assertEqual(result, array([7,15,18,0]))
def test_packArrays(self): """JointEnumeration packArrays should return correct array.""" a = JointEnumeration(['xyz', 'abcd', 'ef']) v = [[0,1,2,0],[3,3,1,0], [1,1,0,0]] result = a.packArrays(v) self.assertEqual(result, array([7,15,18,0]))