From 019e91341fbce90baa3a10a69ed4517b6c2a6168 Mon Sep 17 00:00:00 2001
From: Michael Chin <chnmch@amazon.com>
Date: Wed, 22 Nov 2023 15:11:09 -0800
Subject: [PATCH] add unit tests

---
 test/unit/seed/get_datasets_test.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/test/unit/seed/get_datasets_test.py b/test/unit/seed/get_datasets_test.py
index 2248e41d..2d152032 100644
--- a/test/unit/seed/get_datasets_test.py
+++ b/test/unit/seed/get_datasets_test.py
@@ -71,6 +71,18 @@ def test_get_custom_queries_from_file_sparql(self):
         self.assertEqual(1, len(queries))
         self.assertEqual('0_test_data.txt', queries[0]['name'])
 
+    def test_get_data_sets_cypher(self):
+        data_sets = get_data_sets('opencypher')
+        self.assertTrue('airports' in data_sets)
+
+    def test_get_sample_queries_cypher(self):
+        language = 'opencypher'
+        name = 'airports'
+        location = 'samples'
+        queries = get_queries(language, name, location)
+        self.assertEqual(1, len(queries))
+        self.assertEqual('airports_full.txt', queries[0]['name'])
+
     def test_get_custom_queries_from_directory_cypher(self):
         language = ''
         name = 'local_seed_test_cypher'