def test_create_constraint_query(category): sanitized_category = NeoTransformer.sanitize_category([category]) q = NeoTransformer.create_constraint_query(sanitized_category) assert q == f"CREATE CONSTRAINT ON (n:{sanitized_category}) ASSERT n.id IS UNIQUE"
def test_sanitize_category(): categories = ['biolink:Gene', 'biolink:GeneOrGeneProduct'] s = NeoTransformer.sanitize_category(categories) assert s == ['`biolink:Gene`', '`biolink:GeneOrGeneProduct`']