removed all sql text to services/db/client.py from tests

Author: anngoroshi

test2text/services/db/client.py

@@ -296,3 +296,10 @@ def join_all_tables_by_test_cases(
                             """
         data = self.conn.execute(sql, params)
         return data.fetchall()
+
+    def get_embeddings_by_id(self, id1: int, from_table: str):
+        cursor = self.conn.execute(
+            f"SELECT embedding FROM {from_table} WHERE id = ?",
+            (id1,)
+        )
+        return cursor.fetchone()

tests/test_db/test_tables/test_annotations.py

@@ -67,11 +67,7 @@ def test_set_embedding(self):
             orig_embedding = [0.1] * self.db.annotations.embedding_size
             self.db.annotations.set_embedding(id1, orig_embedding)
             self.db.conn.commit()
-            cursor = self.db.conn.execute(
-                "SELECT embedding FROM Annotations WHERE id = ?", (id1,)
-            )
-            result = cursor.fetchone()
-            cursor.close()
+            result = self.db.get_embeddings_by_id(id1, "Annotations")
             self.assertIsNotNone(result)
             read_embedding = unpack_float32(result[0])
             self.assertEqual(len(read_embedding), self.db.annotations.embedding_size)
@@ -80,11 +76,7 @@ def test_set_embedding(self):
             new_embedding = [0.9] * self.db.annotations.embedding_size
             self.db.annotations.set_embedding(id1, new_embedding)
             self.db.conn.commit()
-            cursor = self.db.conn.execute(
-                "SELECT embedding FROM Annotations WHERE id = ?", (id1,)
-            )
-            result = cursor.fetchone()
-            cursor.close()
+            result = self.db.get_embeddings_by_id(id1, "Annotations")
             self.assertIsNotNone(result)
             read_embedding = unpack_float32(result[0])
             self.assertEqual(len(read_embedding), self.db.annotations.embedding_size)