add check for single sample input with no db provided

j-andrews7 · Nov 16, 2024 · 49e12b7 · 49e12b7
1 parent 92ded6d
commit 49e12b7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/strprofiler/strprofiler.py b/strprofiler/strprofiler.py
@@ -162,11 +162,17 @@ def strprofiler(
 
     samps = df.to_dict(orient="index")
     summaries = []
+
+    # If fewer than 2 samples and no database provided, exit.
+    if len(samps) < 2 and database is None:
+        print("Only 1 sample provided and no database for comparison, exiting.")
+        print("Only 1 sample provided and no database for comparison, exiting.", file=log_file)
+        log_file.close()
+        return
 
     # Database ingress, if present
     # Set 'reference' for subsequent query to either database or inputs all to all
     if database is not None:
-        # load_database(database)
         df_db = utils.str_ingress(
             paths=[database],
             sample_col=sample_col,

diff --git a/tests/ExampleSTR_long_1samp.csv b/tests/ExampleSTR_long_1samp.csv
@@ -0,0 +1,2 @@
+Sample Name,marker1,marker2,marker4,Penta D,Penta E,AMEL
+SampleA,"12, 14",12,"13,13","9,10","12,14",X