class CypherQueriesGeneratorTestCase(unittest.TestCase):
def setUp(self):
schema_filled_file = open(os.path.join("tests", 'schema.yaml'), 'r')
self.keyspace = "playlist"
def test_generate(self):
#filled YAML file analyser
self.cypher_queries_gen = CypherQueriesGenerator(self.keyspace)
nodes = self.cypher_queries_gen.generate()
self.assertEqual(len(nodes), 6)
def test_analyse_csv(self):
# check number of columns on each csv result file
cypher_queries_gen = CypherQueriesGenerator(self.keyspace)
nodes = cypher_queries_gen.generate()
columns = cypher_queries_gen.analyse_csv(["music_results.csv"])
self.assertEqual(columns, [6])
def test_analyse_node(self):
cypher_queries_gen = CypherQueriesGenerator(self.keyspace)
nodes = cypher_queries_gen.generate()
nodes = cypher_queries_gen.analyse_node(["track_by_id"], nodes)
self.assertEqual(len(nodes), 1)
print(nodes[0])
def test_build_queries(self):
#filled YAML file analyser
self.cypher_queries_gen = CypherQueriesGenerator(self.keyspace)
nodes = self.cypher_queries_gen.generate()
self.cypher_queries_gen.build_queries(["track_by_id"], ["music_results.csv"])
cypher_file = open('cypher_', 'r')
statement = cypher_file.read()
path = str(os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))))
self.assertTrue(statement.startswith("LOAD CSV WITH HEADERS"))
def test_build_queries(self):
#filled YAML file analyser
self.cypher_queries_gen = CypherQueriesGenerator(self.keyspace)
nodes = self.cypher_queries_gen.generate()
self.cypher_queries_gen.build_queries(["track_by_id"], ["music_results.csv"])
cypher_file = open('cypher_', 'r')
statement = cypher_file.read()
path = str(os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))))
self.assertTrue(statement.startswith("LOAD CSV WITH HEADERS"))
def test_analyse_node(self): cypher_queries_gen = CypherQueriesGenerator(self.keyspace) nodes = cypher_queries_gen.generate() nodes = cypher_queries_gen.analyse_node(["track_by_id"], nodes) self.assertEqual(len(nodes), 1) print(nodes[0])
def test_analyse_csv(self): # check number of columns on each csv result file cypher_queries_gen = CypherQueriesGenerator(self.keyspace) nodes = cypher_queries_gen.generate() columns = cypher_queries_gen.analyse_csv(["music_results.csv"]) self.assertEqual(columns, [6])
def test_generate(self): #filled YAML file analyser self.cypher_queries_gen = CypherQueriesGenerator(self.keyspace) nodes = self.cypher_queries_gen.generate() self.assertEqual(len(nodes), 6)
