def node_text_embedding_type(request): return flags_parsers.EnumFlag( ggnn_config.NodeTextEmbeddingType, request.param ) @test.Fixture(scope="session", params=testing_databases.GetDatabaseUrls())
def proto_db(opencl_relpaths: List[str]) -> unlabelled_graph_database.Database: """A test fixture which yields a proto database with 256 OpenCL entries.""" with testing_databases.DatabaseContext( unlabelled_graph_database.Database, testing_databases.GetDatabaseUrls()[0]) as db: random_unlabelled_graph_database_generator.PopulateDatabaseWithTestSet( db, len(opencl_relpaths)) yield db
def graph_db(request) -> graph_tuple_database.Database: """A test fixture which returns a graph database with random graphs.""" graph_y_dimensionality, node_y_dimensionality = request.param db = graph_tuple_database.Database(testing_databases.GetDatabaseUrls()[0]) random_graph_tuple_database_generator.PopulateDatabaseWithRandomGraphTuples( db, graph_count=100, graph_y_dimensionality=graph_y_dimensionality, node_y_dimensionality=node_y_dimensionality, ) return db
def graph_db() -> graph_tuple_database.Database: """A test fixture which creates a session-level graph database.""" with testing_databases.DatabaseContext( graph_tuple_database.Database, testing_databases.GetDatabaseUrls()[0] ) as db: random_graph_tuple_database_generator.PopulateDatabaseWithRandomGraphTuples( db, graph_count=20, node_x_dimensionality=2, node_y_dimensionality=0, graph_x_dimensionality=2, graph_y_dimensionality=2, with_data_flow=False, split_count=3, ) yield db
def graph_db( opencl_relpaths: List[str], node_y_dimensionality: int, ) -> graph_tuple_database.Database: """A test fixture which yields a graph database with 100 real graphs.""" with testing_databases.DatabaseContext( graph_tuple_database.Database, testing_databases.GetDatabaseUrls()[0] ) as db: random_graph_tuple_database_generator.PopulateWithTestSet( db, len(opencl_relpaths), node_x_dimensionality=2, node_y_dimensionality=node_y_dimensionality, graph_x_dimensionality=0, graph_y_dimensionality=0, split_count=3, ) yield db
def ir_db(opencl_relpaths: List[str]) -> ir_database.Database: """A test fixture which yields an IR database with 256 OpenCL entries.""" with testing_databases.DatabaseContext( ir_database.Database, testing_databases.GetDatabaseUrls()[0]) as db: rows = [] # Create IRs using OpenCL relpaths. for i, relpath in enumerate(opencl_relpaths): ir = ir_database.IntermediateRepresentation.CreateFromText( source="pact17_opencl_devmap", relpath=relpath, source_language=ir_database.SourceLanguage.OPENCL, type=ir_database.IrType.LLVM_6_0, cflags="", text=CreateRandomString(), ) ir.id = i + 1 rows.append(ir) with db.Session(commit=True) as session: session.add_all(rows) yield db
from deeplearning.ml4pl.testing import random_graph_tuple_generator from deeplearning.ml4pl.testing import random_networkx_generator from deeplearning.ml4pl.testing import testing_databases from labm8.py import decorators from labm8.py import test FLAGS = test.FLAGS ############################################################################### # Fixtures. ############################################################################### @test.Fixture( scope="session", params=testing_databases.GetDatabaseUrls(), namer=testing_databases.DatabaseUrlNamer("graph_db"), ) def empty_db(request) -> graph_tuple_database.Database: """A test fixture which yields an empty graph proto database.""" yield from testing_databases.YieldDatabase(graph_tuple_database.Database, request.param) @test.Fixture( scope="function", params=testing_databases.GetDatabaseUrls(), namer=testing_databases.DatabaseUrlNamer("graph_db"), ) def db_session(request) -> graph_tuple_database.Database.SessionType: """A test fixture which yields an empty graph proto database session."""
from deeplearning.ml4pl.models.ggnn import ggnn_config >>>>>>> de933d07a... Add a node text embedding enum.:deeplearning/ml4pl/models/ggnn/ggnn_test.py from deeplearning.ml4pl.testing import random_graph_tuple_database_generator from deeplearning.ml4pl.testing import testing_databases FLAGS = test.FLAGS ############################################################################### # Fixtures. ############################################################################### @test.Fixture( scope="session", params=testing_databases.GetDatabaseUrls(), namer=testing_databases.DatabaseUrlNamer("log_db"), ) def log_db(request) -> log_database.Database: """A test fixture which yields an empty log database.""" yield from testing_databases.YieldDatabase( log_database.Database, request.param ) @test.Fixture(scope="session") def logger(log_db: log_database.Database) -> logging.Logger: """A test fixture which yields a logger.""" with logging.Logger(log_db, max_buffer_length=128) as logger: yield logger