This notebook was prepared by [Donne Martin](http://donnemartin.com). Source and license info is on [GitHub](https://github.com/donnemartin/interactive-coding-challenges).

Challenge Notebook

Problem: Implement quick sort.

Constraints
Test Cases
Algorithm
Code
Unit Test
Solution Notebook

Constraints

Is a naiive solution sufficient (ie not in-place)?
- Yes

Test Cases

Empty input -> []
One element -> [element]
Two or more elements

Algorithm

Refer to the Solution Notebook. If you are stuck and need a hint, the solution notebook's algorithm discussion might be a good place to start.

Code



In [ ]:

    
def quick_sort(data):
    # TODO: Implement me
    pass

Unit Test

The following unit test is expected to fail until you solve the challenge.



In [ ]:

    
# %load test_quick_sort.py
from nose.tools import assert_equal


class TestQuickSort(object):
    def test_quick_sort(self, func):
        print('Empty input')
        data = []
        sorted_data = func(data)
        assert_equal(sorted_data, [])

        print('One element')
        data = [5]
        sorted_data = func(data)
        assert_equal(sorted_data, [5])

        print('Two or more elements')
        data = [5, 1, 7, 2, 6, -3, 5, 7, -1]
        sorted_data = func(data)
        assert_equal(sorted_data, sorted(data))

        print('Success: test_quick_sort\n')


def main():
    test = TestQuickSort()
    test.test_quick_sort(quick_sort)
    try:
        test.test_quick_sort(quick_sort_alt)
    except NameError:
        # Alternate solutions are only defined
        # in the solutions file
        pass


if __name__ == '__main__':
    main()

Solution Notebook

Review the Solution Notebook for a discussion on algorithms and code solutions.