run_flag is optional again in sqlite init, updated test_sqlite.py

lanl · Nov 27, 2024 · 0ff5e5b · 0ff5e5b
1 parent ec7ade5
commit 0ff5e5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/dsi/backends/sqlite.py b/dsi/backends/sqlite.py
@@ -42,7 +42,7 @@ class Sqlite(Filesystem):
     con = None
     cur = None
 
-    def __init__(self, filename, run_table):
+    def __init__(self, filename, run_table = True):
         self.filename = filename
         self.con = sqlite3.connect(filename)
         self.cur = self.con.cursor()

diff --git a/dsi/plugins/tests/test_file_reader.py b/dsi/plugins/tests/test_file_reader.py
@@ -23,10 +23,9 @@ def test_bueno_plugin_adds_rows():
     path2 = '/'.join([get_git_root('.'), 'examples/data', 'bueno2.data'])
     plug = Bueno(filenames=[path1, path2])
     plug.add_rows()
-    plug.add_rows()
 
     for key, val in plug.output_collector["Bueno"].items():
-        assert len(val) == 4  # two lists of length 4
+        assert len(val) == 2  # two lists of length 4
 
     # 4 Bueno cols
     assert len(plug.output_collector["Bueno"].keys()) == 4