def test_time_grain_addons(self): with app.app_context(): app.config["TIME_GRAIN_ADDONS"] = {"PTXM": "x seconds"} app.config["TIME_GRAIN_ADDON_FUNCTIONS"] = { "sqlite": {"PTXM": "ABC({col})"} } time_grains = SqliteEngineSpec.get_time_grains() time_grain_addon = time_grains[-1] self.assertEqual("PTXM", time_grain_addon.duration) self.assertEqual("x seconds", time_grain_addon.label)
def test_time_grain_addons(): config = app.config.copy() app.config["TIME_GRAIN_ADDONS"] = {"PTXM": "x seconds"} app.config["TIME_GRAIN_ADDON_EXPRESSIONS"] = { "sqlite": { "PTXM": "ABC({col})" } } with app.app_context(): time_grains = SqliteEngineSpec.get_time_grains() time_grain_addon = time_grains[-1] assert "PTXM" == time_grain_addon.duration assert "x seconds" == time_grain_addon.label app.config = config