def test_filter_specs(self): delegate_spec = DimensionSpec('dim', 'out').build() filter_specs = [ (ListFilteredSpec(['val1', 'val2']), { 'type': 'listFiltered', 'delegate': delegate_spec, 'values': ['val1', 'val2'], }), (ListFilteredSpec(['val1', 'val2'], is_whitelist=False), { 'type': 'listFiltered', 'delegate': delegate_spec, 'values': ['val1', 'val2'], 'isWhitelist': False, }), (RegexFilteredSpec(r'\w+'), { 'type': 'regexFiltered', 'delegate': delegate_spec, 'pattern': '\\w+', }) ] for filter_spec, expected_dim_spec in filter_specs: dim_spec = DimensionSpec('dim', 'out', filter_spec=filter_spec) actual = dim_spec.build() assert actual == expected_dim_spec
def test_filter_specs(self): delegate_spec = DimensionSpec("dim", "out").build() filter_specs = [ ( ListFilteredSpec(["val1", "val2"]), { "type": "listFiltered", "delegate": delegate_spec, "values": ["val1", "val2"], }, ), ( ListFilteredSpec(["val1", "val2"], is_whitelist=False), { "type": "listFiltered", "delegate": delegate_spec, "values": ["val1", "val2"], "isWhitelist": False, }, ), ( RegexFilteredSpec(r"\w+"), { "type": "regexFiltered", "delegate": delegate_spec, "pattern": "\\w+" }, ), ] for filter_spec, expected_dim_spec in filter_specs: dim_spec = DimensionSpec("dim", "out", filter_spec=filter_spec) actual = dim_spec.build() assert actual == expected_dim_spec
def test_filter_specs(self): delegate_spec = DimensionSpec('dim', 'out').build() filter_specs = [(ListFilteredSpec(['val1', 'val2']), { 'type': 'listFiltered', 'delegate': delegate_spec, 'values': ['val1', 'val2'], }), (ListFilteredSpec(['val1', 'val2'], is_whitelist=False), { 'type': 'listFiltered', 'delegate': delegate_spec, 'values': ['val1', 'val2'], 'isWhitelist': False, }), (RegexFilteredSpec(r'\w+'), { 'type': 'regexFiltered', 'delegate': delegate_spec, 'pattern': '\\w+', })] for filter_spec, expected_dim_spec in filter_specs: dim_spec = DimensionSpec('dim', 'out', filter_spec=filter_spec) actual = dim_spec.build() assert actual == expected_dim_spec
def test_extraction_functions(self): js_func = 'function(x) {return x};' ext_fns = [ (RegexExtraction(r'\w+'), {'type': 'regex', 'expr': '\\w+'}), (PartialExtraction(r'\w+'), {'type': 'partial', 'expr': '\\w+'}), (JavascriptExtraction(js_func), { 'type': 'javascript', 'function': js_func, 'injective': False }), (MapLookupExtraction(TestMapLookupExtraction.mapping), { 'type': 'lookup', 'lookup': { 'type': 'map', 'map': TestMapLookupExtraction.mapping }, 'retainMissingValue': False, 'replaceMissingValueWith': None, 'injective': False }) ] for ext_fn, expected_ext_fn in ext_fns: dim_spec = DimensionSpec('dim', 'out', extraction_function=ext_fn) actual = dim_spec.build() expected = { 'type': 'extraction', 'dimension': 'dim', 'outputName': 'out', 'extractionFn': expected_ext_fn } assert actual == expected
def test_extraction_functions(self): js_func = "function(x) {return x};" ext_fns = [ (RegexExtraction(r"\w+"), { "type": "regex", "expr": "\\w+" }), (PartialExtraction(r"\w+"), { "type": "partial", "expr": "\\w+" }), ( JavascriptExtraction(js_func), { "type": "javascript", "function": js_func, "injective": False }, ), ( MapLookupExtraction(TestMapLookupExtraction.mapping), { "type": "lookup", "lookup": { "type": "map", "map": TestMapLookupExtraction.mapping }, "retainMissingValue": False, "replaceMissingValueWith": None, "injective": False, }, ), ] for ext_fn, expected_ext_fn in ext_fns: dim_spec = DimensionSpec("dim", "out", extraction_function=ext_fn) actual = dim_spec.build() expected = { "type": "extraction", "dimension": "dim", "outputName": "out", "extractionFn": expected_ext_fn, } assert actual == expected
def test_extraction_functions(self): js_func = "function(x) {return x};" ext_fns = [ (RegexExtraction(r"\w+"), {"type": "regex", "expr": "\\w+"}), (PartialExtraction(r"\w+"), {"type": "partial", "expr": "\\w+"}), (JavascriptExtraction(js_func), {"type": "javascript", "function": js_func, "injective": False}), ( MapLookupExtraction(TestMapLookupExtraction.mapping), { "type": "lookup", "lookup": {"type": "map", "map": TestMapLookupExtraction.mapping}, "retainMissingValue": False, "replaceMissingValueWith": None, "injective": False, }, ), ] for ext_fn, expected_ext_fn in ext_fns: dim_spec = DimensionSpec("dim", "out", extraction_function=ext_fn) actual = dim_spec.build() expected = {"type": "extraction", "dimension": "dim", "outputName": "out", "extractionFn": expected_ext_fn} assert actual == expected
def test_default(self): dim_spec = DimensionSpec("dim", "out") actual = dim_spec.build() expected = {"type": "default", "dimension": "dim", "outputName": "out"} assert actual == expected
def test_build_dimension(self): assert build_dimension("raw_dim") == "raw_dim" dim_spec = DimensionSpec("dim", "out") assert build_dimension(dim_spec) == dim_spec.build()
def test_build_dimension(self): assert build_dimension('raw_dim') == 'raw_dim' dim_spec = DimensionSpec('dim', 'out') assert build_dimension(dim_spec) == dim_spec.build()
def test_default(self): dim_spec = DimensionSpec('dim', 'out') actual = dim_spec.build() expected = {'type': 'default', 'dimension': 'dim', 'outputName': 'out'} assert actual == expected