Initial commit for metrik

test/targets/pickle_target.py

@@ -0,0 +1,40 @@
+from unittest import TestCase
+import luigi
+from metrik.targets.pickle_target import PickleTarget
+
+
+class FibTask(luigi.Task):
+    s = luigi.IntParameter()
+
+    def requires(self):
+        if self.s >= 2:
+            return [FibTask(self.s - 1), FibTask(self.s - 2)]
+        else:
+            return []
+
+    def output(self):
+        return PickleTarget(self.task_id)
+
+    def run(self):
+        if self.s <= 1:
+            val = self.s
+        else:
+            count = 0
+            for input in self.input():
+                count += input.read()
+            val = count
+
+        self.output().write(val)
+
+
+class TestPickleTarget(TestCase):
+    def test_fibonacci(self):
+        f = FibTask(6)
+        luigi.build([f], local_scheduler=True)
+
+        ret = f.output().read()
+        assert ret == 8
+
+        f = FibTask(100)
+        luigi.build([f], local_scheduler=True)
+        assert f.output().read() == 354224848179261915075

test/tasks/test_bloomberg.py

@@ -0,0 +1,22 @@
+from unittest import TestCase
+
+from metrik.conf import USER_AGENT
+from metrik.tasks.bloomberg import BloombergEquityInfo
+
+
+class BloombergTest(TestCase):
+    def test_correct_info_apple(self):
+        sector, industry, sub_industry = \
+            BloombergEquityInfo.retrieve_info("AAPL:US", USER_AGENT)
+
+        assert sector == 'Technology'
+        assert industry == 'Hardware'
+        assert sub_industry == 'Communications Equipment'
+
+    def test_correct_info_kcg(self):
+        sector, industry, sub_industry = \
+            BloombergEquityInfo.retrieve_info("KCG:US", USER_AGENT)
+
+        assert sector == 'Financials'
+        assert industry == 'Institutional Financial Svcs'
+        assert sub_industry == 'Institutional Brokerage'