Add example code and related end-to-end tests.

1 files changed, 67 insertions, 0 deletions

diff --git a/tests/example_end_to_end.py b/tests/example_end_to_end.py
new file mode 100644
index 0000000..1181490
--- /dev/null
+++ b/tests/example_end_to_end.py
@@ -0,0 +1,67 @@
+# Uses the example code in the example/ directory, since it's a good end-to-end
+# test.
+from __future__ import division
+
+import unittest
+
+from example.driver import get_script_dir, nolock_example, lock_example
+
+nolock_example_1_list = [0, 55575, 55575, 67570, 67569, 8805, 8805, 68149,
+                         68149, 12514, 12513, 55400, 55400, 76501, 76501,
+                         66761, 66762, -324, -324, 91995, 91995, 35996, 35997]
+
+nolock_example_2_list = None
+lock_example_1_list = [-44967, -104360, -9607, -44134, -53237, -31791, -68420,
+                       -67526, -55107, -47183, -11898]
+
+lock_example_2_list = None
+
+class NoLockExample(unittest.TestCase):
+    def setUp(self):
+        example_dir = get_script_dir()
+        self.nolock_example_1 = example_dir + "/nolock_example_1"
+        self.nolock_example_2 = example_dir + "/nolock_example_2"
+        results = nolock_example([self.nolock_example_1, self.nolock_example_2])
+        self.distilled = []
+        for (name, clusts) in results:
+            if clusts is None:
+                self.distilled.append((name, None))
+            else:
+                new_distilled = []
+                for clust in clusts:
+                    new_distilled += [task.response_time - task.deadline
+                                     for task in clust]
+                self.distilled.append((name, new_distilled))
+
+    def test_nolock_example_1_works(self):
+        self.assertEqual(self.distilled[0], (self.nolock_example_1,
+                                             nolock_example_1_list))
+
+    def test_nolock_example_2_works(self):
+        self.assertEqual(self.distilled[1], (self.nolock_example_2,
+                                             nolock_example_2_list))
+
+class LockExample(unittest.TestCase):
+    def setUp(self):
+        example_dir = get_script_dir()
+        self.lock_example_1 = example_dir + "/lock_example_1"
+        self.lock_example_2 = example_dir + "/lock_example_2"
+        results = lock_example([self.lock_example_1, self.lock_example_2])
+        self.distilled = []
+        for (name, clusts) in results:
+            if clusts is None:
+                self.distilled.append((name, None))
+            else:
+                new_distilled = []
+                for clust in clusts:
+                    new_distilled += [task.response_time - task.deadline
+                                     for task in clust]
+                self.distilled.append((name, new_distilled))
+
+    def test_nolock_example_1_works(self):
+        self.assertEqual(self.distilled[0], (self.lock_example_1,
+                                             lock_example_1_list))
+
+    def test_nolock_example_2_works(self):
+        self.assertEqual(self.distilled[1], (self.lock_example_2,
+                                             lock_example_2_list))