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
