def test_simple_exports():
"""
check we can understand a simple variable assignment
"""
testdata = {"code": "x = 5", "frames": [], "hash": "doesnmatter"}
result = handle_exports(testdata)
print(result)
assert (result["exports"][0] == "x")
def test_non_assignment_imports():
"""
variables can be used in statements that are not assignments..
"""
code = "a.dosomething(b,c)" # a, b and c should be imports
testdata = {"code": code, "frames": ["a", "b", "c"], "hash": "irrelevant"}
result = handle_exports(testdata)
print(result)
assert (sorted(result["imports"]) == ["a", "b", "c"])
def test_simple_imports():
"""
check we can understand a simple variable assignment where
one term on the rhs is an existing frame
"""
testdata = {"code": "x = y + 5", "frames": ["y"], "hash": "doesnmatter"}
result = handle_exports(testdata)
print(result)
assert (result["imports"][0] == "y")
def test_imports_used_in_functions():
"""
imported variables can be used inside functions.
"""
code = 'def hello():\n x = a + b\n return x\n\nnewx = hello()\n\n'
testdata = {"code": code, "frames": ["a", "b"], "hash": "irrelevant"}
result = handle_exports(testdata)
print(result)
assert (result["exports"] == ["newx"])
assert (sorted(result["imports"]) == ["a", "b"])
def test_out_of_scope_assignments():
"""
variables assigned e.g. inside function definitions should not
be added to exports
"""
code = 'def hello():\n x = pd.DataFrame({"xx":[1,2,3]})\n return x\n\nnewx = hello()\n\n'
testdata = {"code": code, "frames": [], "hash": "irrelevant"}
result = handle_exports(testdata)
print(result)
assert (result["exports"] == ["newx"])
def test_exports_same_var_twice():
"""
We only want to upload a dataframe once, even if
the variable appears in the code fragment multiple times.
"""
code = "x = a + b\nx = c + d\n"
testdata = {"code": code, "frames": ["a", "b", "c"], "hash": "irrelevant"}
result = handle_exports(testdata)
print(result)
assert (result["exports"].count("x") == 1)
def test_confusing_names():
"""
Need to tell the difference between e.g. "x" and "xx"
"""
testdata = {
"code": "xx = xxx(x) + yy",
"frames": ["x", "y"],
"hash": "doesntmatter"
}
result = handle_exports(testdata)
print(result)
assert (result["exports"] == ["xx"])
assert (result["imports"] == ["x"])
def test_multiple_assignment():
"""
function might have two return values to unpack
"""
testdata = {
"code": "x,y = funcoutput(z)",
"frames": ["z", "a", "b"],
"hash": "doesntmatter"
}
result = handle_exports(testdata)
print(result)
assert (sorted(result["exports"]) == ["x", "y"])
assert (result["imports"] == ["z"])
def test_subscript_imports_exports():
"""
If we look at an element of an object such as a dataframe(e.g. x[0]), that dataframe needs to be
defined so should be an import.
If the element is an assignment target it should also be an export.
"""
code = "df.x[0] = 3\n"
frames = ["df"]
testdata = {"code": code, "frames": frames, "hash": "irrelevant"}
result = handle_exports(testdata)
assert (len(result["imports"]) == 1)
assert (result["imports"][0] == "df")
assert (len(result["exports"]) == 1)
assert (result["exports"][0] == "df")
def exports():
data = json.loads(request.data.decode("utf-8"))
imports_exports = handle_exports(data)
return jsonify(imports_exports)
