diff options
Diffstat (limited to 'challenge-113/roger-bell-west/python/ch-2.py')
| -rwxr-xr-x | challenge-113/roger-bell-west/python/ch-2.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/challenge-113/roger-bell-west/python/ch-2.py b/challenge-113/roger-bell-west/python/ch-2.py new file mode 100755 index 0000000000..48a01cb2b0 --- /dev/null +++ b/challenge-113/roger-bell-west/python/ch-2.py @@ -0,0 +1,23 @@ +#! /usr/bin/python3 + +import unittest + +def rbt(ti): + s=0 + for n in ti: + if n>0: + s += n + to=list() + for n in ti: + if n>0: + to.append(s-n) + else: + to.append(n) + return to + +class TestRbt(unittest.TestCase): + + def test_ex1(self): + self.assertEqual(rbt([1,2,3,4,-1,5,6,-1,7]),[27,26,25,24,-1,23,22,-1,21],'example 1') + +unittest.main() |
