def testSortedStride(self):
for inp in self.inputs:
weights = [(20 - i) for i in inp[0]]
pfunc = partition.SortedStride()
actual = pfunc(list(zip(inp[0], weights)), inp[1], inp[2])
expected = list(inp[0][:])
expected.reverse()
expected = expected[inp[1]::inp[2]]
self.assertEqual(actual, expected)
def testSortedStride(self):
for inp in self.inputs:
weights = array([(20 - i) for i in inp[0]])
pfunc = partition.SortedStride()
data = dstack((inp[0], weights))[0]
actual = pfunc(data, inp[1], inp[2])
expected = inp[0][::-1]
expected = expected[inp[1]::inp[2]]
testing.assert_array_equal(actual, expected)
def testSortedStride(self):
for inp in self.inputs:
weights = numpy.array([(20 - i) for i in inp[0]])
pfunc = partition.SortedStride()
data = numpy.dstack((inp[0], weights))[0]
actual = pfunc(data, inp[1], inp[2])
expected = inp[0][::-1]
expected = expected[inp[1]::inp[2]]
msg = test_info_msg('SortedStride', data, inp[1], inp[2], actual,
expected)
print msg
numpy.testing.assert_array_equal(actual, expected, msg)
def testSortedStride(self):
for inp in self.inputs:
weights = [(20 - i) for i in inp[0]]
pfunc = partition.SortedStride()
actual = pfunc(zip(inp[0], weights), inp[1], inp[2])
expected = inp[0][:]
expected.reverse()
expected = expected[inp[1]::inp[2]]
msg = test_info_msg('SortedStride', zip(inp[0], weights), inp[1],
inp[2], actual, expected)
print msg
self.assertEqual(actual, expected, msg)