mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
|
from nose.tools import assert_equal
|
||
|
|
|
||
|
|
|
||
|
|
class TestPowerSet(object):
|
||
|
|
|
||
|
|
def test_power_set(self):
|
||
|
|
input_set = ''
|
||
|
|
expected = ['']
|
||
|
|
self.run_test(input_set, expected)
|
||
|
|
input_set = 'a'
|
||
|
|
expected = ['a', '']
|
||
|
|
self.run_test(input_set, expected)
|
||
|
|
input_set = 'ab'
|
||
|
|
expected = ['a', 'ab', 'b', '']
|
||
|
|
self.run_test(input_set, expected)
|
||
|
|
input_set = 'abc'
|
||
|
|
expected = ['a', 'ab', 'abc', 'ac',
|
||
|
|
'b', 'bc', 'c', '']
|
||
|
|
self.run_test(input_set, expected)
|
||
|
|
input_set = 'aabc'
|
||
|
|
expected = ['a', 'aa', 'aab', 'aabc',
|
||
|
|
'aac', 'ab', 'abc', 'ac',
|
||
|
|
'b', 'bc', 'c', '']
|
||
|
|
self.run_test(input_set, expected)
|
||
|
|
print('Success: test_power_set')
|
||
|
|
|
||
|
|
def run_test(self, input_set, expected):
|
||
|
|
combinatoric = Combinatoric()
|
||
|
|
result = combinatoric.find_power_set(input_set)
|
||
|
|
assert_equal(result, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
test = TestPowerSet()
|
||
|
|
test.test_power_set()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|