コード例 #1
0
ファイル: test_sort.py プロジェクト: weedySeaDragon/ivalice
def test_quicksort():
    rng = np.random.RandomState(0)
    values = rng.rand(500)
    indices = np.arange(len(values)).astype(np.int32)

    sorted_idx = np.argsort(values)
    sorted_values = values[sorted_idx]
    sorted_indices = indices[sorted_idx]

    quicksort(values, indices, 0, len(values) - 1)

    assert_array_equal(sorted_values, values)
    assert_array_equal(sorted_indices, indices)
コード例 #2
0
ファイル: test_sort.py プロジェクト: weedySeaDragon/ivalice
def test_quicksort_one():
    values = np.arange(1).astype(np.float64)
    indices = np.arange(1).astype(np.int32)
    quicksort(values, indices, 0, len(values) - 1)