def test_query_engine_params(self): from luiti.daemon.query_engine.params import Params default_query = {"date_value": "2014-09-01", "language": "Chinese", "gender": "Male"} selected_query = {"date_value": "2015-07-02", "city": ["Beijing", "London"]} params_array = Params.build_params_array(default_query, selected_query) base_expected_opt = { "date_value": ArrowParameter.get("2015-07-02"), "city": ["Beijing", "London"], "language": "Chinese", "gender": "Male", } expected_opts = [ dict(base_expected_opt.items() + {"city": "Beijing"}.items()), dict(base_expected_opt.items() + {"city": "London"}.items()), ] self.assertEqual(params_array, expected_opts)
def test_utils(self): from luiti.daemon.utils import TaskStorageSet, TaskStorageDict current_time = ArrowParameter.now() t1 = FoobarDay(date_value=current_time, foobar="1") t2 = FoobarDay(date_value=current_time, foobar="1") s1 = TaskStorageSet() s1.add(t1) self.assertEqual(len(s1), 1) s1.add(t2) self.assertEqual(len(s1), 1) self.assertTrue(t1 in s1) self.assertTrue(t2 in s1) d1 = TaskStorageDict() d1[t1].add(t1) d1[t2].add(t2) self.assertEqual(len(d1), 1) self.assertEqual(len(d1[t1]), 1)
# -*- coding: utf-8 -*- import os import sys root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.insert(0, root_dir) import unittest from luiti.task_templates import TaskDay from luiti import luigi from luiti.luigi_extensions import ArrowParameter current_time = ArrowParameter.now() from luiti.tests import SetupLuitiPackages, date_begin config = SetupLuitiPackages.config class FoobarDay(TaskDay): root_dir = "/tmp" foobar = luigi.Parameter() class TestDaemon(unittest.TestCase): def test_Server(self): from luiti.daemon import Server s1 = Server("0.0.0.0", 8082) self.assertEqual(s1.url, "http://localhost:8082") self.assertTrue(s1.app)