def test_js_not_injective(self):
js_func = "function(x) {return x};"
ext_fn = JavascriptExtraction(js_func)
actual = ext_fn.build()
expected = {"type": "javascript", "function": js_func, "injective": False}
assert actual == expected
def test_js_not_injective(self):
js_func = "function(x) {return x};"
ext_fn = JavascriptExtraction(js_func)
actual = ext_fn.build()
expected = {
"type": "javascript",
"function": js_func,
"injective": False
}
assert actual == expected
def test_js_not_injective(self):
js_func = 'function(x) {return x};'
ext_fn = JavascriptExtraction(js_func)
actual = ext_fn.build()
expected = {
'type': 'javascript',
'function': js_func,
'injective': False
}
assert actual == expected
def test_js_not_injective(self):
js_func = 'function(x) {return x};'
ext_fn = JavascriptExtraction(js_func)
actual = ext_fn.build()
expected = {
'type': 'javascript',
'function': js_func,
'injective': False
}
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