aboutsummaryrefslogtreecommitdiff
path: root/challenge-245/roger-bell-west/python/ch-1.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-245/roger-bell-west/python/ch-1.py')
-rwxr-xr-xchallenge-245/roger-bell-west/python/ch-1.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/challenge-245/roger-bell-west/python/ch-1.py b/challenge-245/roger-bell-west/python/ch-1.py
new file mode 100755
index 0000000000..f2714b0bc2
--- /dev/null
+++ b/challenge-245/roger-bell-west/python/ch-1.py
@@ -0,0 +1,18 @@
+#! /usr/bin/python3
+
+def sortlanguage(langs, popularities):
+ ix = list(range(len(langs)))
+ ix.sort(key = lambda n: popularities[n])
+ return [langs[n] for n in ix]
+
+import unittest
+
+class TestSortlanguage(unittest.TestCase):
+
+ def test_ex1(self):
+ self.assertEqual(sortlanguage(["perl", "c", "python"], [2, 1, 3]), ["c", "perl", "python"], 'example 1')
+
+ def test_ex2(self):
+ self.assertEqual(sortlanguage(["c++", "haskell", "java"], [1, 3, 2]), ["c++", "java", "haskell"], 'example 2')
+
+unittest.main()