Extract a strain name given a csv string and it's header

* gn3/csvcmp.py (extract_strain_name): New function.
* gn3/db/sample_data (delete_sample_data): Use the aforementioned function.
(insert_sample_data): Ditto.
* tests/unit/test_csvcmp: Test cases for above.

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/unit/test_csvcmp.py b/tests/unit/test_csvcmp.py
index 4a96f99..fd7aa28 100644
--- a/tests/unit/test_csvcmp.py
+++ b/tests/unit/test_csvcmp.py
@@ -1,6 +1,7 @@
 from gn3.csvcmp import csv_diff
 from gn3.csvcmp import fill_csv
 from gn3.csvcmp import remove_insignificant_edits
+from gn3.csvcmp import extract_strain_name
 
 import pytest
 
@@ -99,3 +100,10 @@ BXD15,14,x,x
                         'Additions': [],
                         'Deletions': [],
                         'Modifications': []})
+
+
+@pytest.mark.unit_test
+def test_extract_strain_name():
+    assert(extract_strain_name(csv_header="Strain Name,Value,SE,Count",
+                               data="BXD1,18,x,0") ==
+           "BXD1")