From b6ecd0888ccf51b1dd2bfcbbbc4e03b1c5a92699 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 1 Mar 2022 15:41:56 +0300
Subject: 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.
---
 tests/unit/test_csvcmp.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'tests/unit')

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")
-- 
cgit v1.2.3