-
Notifications
You must be signed in to change notification settings - Fork 0
/
testmeta.py
49 lines (34 loc) · 1.2 KB
/
testmeta.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import sonnixgres
def test_metadata_enumeration():
# Create a PostgresCredentials object
credentials = sonnixgres.PostgresCredentials()
# Create a MetadataCache object
metadata_cache = sonnixgres.MetadataCache(credentials.schema, credentials.tables)
# Refresh the metadata cache
metadata_cache.refresh_metadata_cache()
# Display the metadata cache
metadata_cache.display_metadata()
def test_query_and_display():
# Create a psycopg2 connection
connection = sonnixgres.create_connection()
# Define SQL query
query = """
SELECT manufacturer, SUM(active_minutes) AS total_active_minutes, AVG(price) AS avg_price
FROM ai.underutilized_users_ams
GROUP BY manufacturer
ORDER BY avg_price DESC, total_active_minutes DESC
LIMIT 10;
"""
# Execute the query
result_df = sonnixgres.query_database(connection, query)
# Display the results as a table
sonnixgres.display_results_as_table(result_df)
# Close the connection
connection.close()
def main():
print("Testing Metadata Enumeration...")
test_metadata_enumeration()
print("\nTesting Query Execution and Display...")
test_query_and_display()
if __name__ == "__main__":
main()