# Copyright(c) Microsoft Corporation.
# Licensed under the MIT license.
import sqlmlutils
def foo():
return "bar"
sqlpython = sqlmlutils.SQLPythonExecutor(
sqlmlutils.ConnectionInfo(server="localhost", database="master"))
result = sqlpython.execute_function_in_sql(foo)
assert result == "bar"
# Copyright(c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT license.
import pytest
import sqlmlutils
from contextlib import redirect_stdout
from subprocess import Popen, PIPE, STDOUT
from pandas import DataFrame
import io
import os
from conftest import connection
current_dir = os.path.dirname(__file__)
script_dir = os.path.join(current_dir, "scripts")
sqlpy = sqlmlutils.SQLPythonExecutor(connection)
###################
# No output tests #
###################
def test_no_output():
def my_func():
print("blah blah blah")
name = "test_no_output"
sqlpy.drop_sproc(name)
sqlpy.create_sproc_from_function(name, my_func)
assert sqlpy.check_sproc(name)
import matplotlib.pyplot as plt
import io
title = x_col + " vs. " + y_col
plt.scatter(input_df[x_col], input_df[y_col])
plt.xlabel(x_col)
plt.ylabel(y_col)
plt.title(title)
# Save scatter plot image as a png
buf = io.BytesIO()
plt.savefig(buf, format="png")
buf.seek(0)
# Returns the bytes of the png to the client
return buf
sqlpy = sqlmlutils.SQLPythonExecutor(
sqlmlutils.ConnectionInfo(server="localhost", database="AirlineTestDB"))
sql_query = "select top 100 * from airline5000"
plot_data = sqlpy.execute_function_in_sql(func=scatter_plot,
input_data_query=sql_query,
x_col="ArrDelay",
y_col="CRSDepTime")
im = Image.open(plot_data)
im.show()
#im.save("scatter_test.png")
