Untitled

"""
This module tests the database class and its functionalities
"""
from google.cloud.spanner_v1.types import Type, TypeCode, StructType
from database import SpannerDatabase
from conversion import columns_to_native_numpy, prepare_data_for_graphing

def main():
    # """Test that the SpannerDatabase class is initialized correctly"""
    db = SpannerDatabase("gvizlab", "vizlab", "graphcast")

    simple_query = "GRAPH FinGraph MATCH (p:Person)-[:OWNS]->(acc:Account) RETURN TO_JSON(p) as person_json_single, TO_JSON(acc) as account_json_single;"
    simple_query_result, simple_fields = db.execute_query(simple_query)
    simple_d, simple_ignored_columns = columns_to_native_numpy(simple_query_result, simple_fields)

    complex_query = "GRAPH FinGraph MATCH  (synthesized_id:Person {id: 5})-[:OWNS]->(:Account)  (()-[transfer:Transfers]->(layering_accnt)){1,4}  (:Account)<-[:OWNS]-(another_synthesized_id:Person {id: 9}) RETURN TO_JSON(layering_accnt) AS layering_accnts, TO_JSON(transfer) AS transfers LIMIT 5"
    complex_query_result, complex_fields = db.execute_query(complex_query)
    complex_d, complex_gnored_columns = columns_to_native_numpy(complex_query_result, complex_fields)
    
    simple_graph = prepare_data_for_graphing(simple_d, False)
    complex_graph = prepare_data_for_graphing(complex_d, False)
    print("Simple Graph: ", simple_graph)
    print("Complex Graph: ", complex_graph)

if __name__ == "__main__":
    main()