diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/unit/db/test_sample_data.py | 18 | ||||
| -rw-r--r-- | tests/unit/test_csvcmp.py | 54 |
2 files changed, 55 insertions, 17 deletions
diff --git a/tests/unit/db/test_sample_data.py b/tests/unit/db/test_sample_data.py index 2524e07..607b278 100644 --- a/tests/unit/db/test_sample_data.py +++ b/tests/unit/db/test_sample_data.py @@ -34,11 +34,13 @@ def test_insert_sample_data(mocker): ) calls = [ mocker.call( - "SELECT Id FROM PublishData where Id = %s " "AND StrainId = %s", + "SELECT Id FROM PublishData where Id = %s " + "AND StrainId = %s", (data_id, strain_id), ), mocker.call( - "INSERT INTO PublishData " "(StrainId, Id, value) VALUES (%s, %s, %s)", + "INSERT INTO PublishData " + "(StrainId, Id, value) VALUES (%s, %s, %s)", (strain_id, data_id, "18"), ), mocker.call( @@ -47,10 +49,13 @@ def test_insert_sample_data(mocker): (strain_id, data_id, "3"), ), mocker.call( - "INSERT INTO NStrain " "(StrainId, DataId, count) VALUES (%s, %s, %s)", + "INSERT INTO NStrain " + "(StrainId, DataId, count) VALUES (%s, %s, %s)", (strain_id, data_id, "0"), ), - mocker.call("SELECT Id FROM CaseAttribute WHERE Name = %s", ("Sex",)), + mocker.call( + "SELECT Id FROM CaseAttribute WHERE Name = %s", ("Sex",) + ), mocker.call( "SELECT StrainId FROM CaseAttributeXRefNew " "WHERE StrainId = %s AND " @@ -135,7 +140,10 @@ def test_extract_actions(): ) == { "delete": None, "insert": {"data": "BXD1,2,F", "csv_header": "Strain Name,SE,Sex"}, - "update": {"data": "BXD1,19,1", "csv_header": "Strain Name,Value,Count"}, + "update": { + "data": "BXD1,19,1", + "csv_header": "Strain Name,Value,Count", + }, } diff --git a/tests/unit/test_csvcmp.py b/tests/unit/test_csvcmp.py index c2fda6b..0843bef 100644 --- a/tests/unit/test_csvcmp.py +++ b/tests/unit/test_csvcmp.py @@ -82,7 +82,9 @@ BXD15,14,x,x""" "Additions": [], "Columns": "Strain Name,Value,SE,Count,Sex", "Deletions": [], - "Modifications": [{"Current": "BXD12,16,x,x,1", "Original": "BXD12,16,x,x,x"}], + "Modifications": [ + {"Current": "BXD12,16,x,x,1", "Original": "BXD12,16,x,x,x"} + ], } @@ -113,7 +115,9 @@ BXD15,14,x,x def test_extract_strain_name(): """Test that the strain's name is extracted given a csv header""" assert ( - extract_strain_name(csv_header="Strain Name,Value,SE,Count", data="BXD1,18,x,0") + extract_strain_name( + csv_header="Strain Name,Value,SE,Count", data="BXD1,18,x,0" + ) == "BXD1" ) @@ -123,27 +127,53 @@ def test_get_allowable_csv_headers(mocker): """Test that all the csv headers are fetched properly""" mock_conn = mocker.MagicMock() expected_values = [ - "Strain Name", "Value", "SE", "Count", - "Condition", "Tissue", "Sex", "Age", - "Ethn.", "PMI (hrs)", "pH", "Color", + "Strain Name", + "Value", + "SE", + "Count", + "Condition", + "Tissue", + "Sex", + "Age", + "Ethn.", + "PMI (hrs)", + "pH", + "Color", ] with mock_conn.cursor() as cursor: cursor.fetchall.return_value = ( - ('Condition',), ('Tissue',), ('Sex',), - ('Age',), ('Ethn.',), ('PMI (hrs)',), ('pH',), ('Color',)) + ("Condition",), + ("Tissue",), + ("Sex",), + ("Age",), + ("Ethn.",), + ("PMI (hrs)",), + ("pH",), + ("Color",), + ) assert get_allowable_sampledata_headers(mock_conn) == expected_values cursor.execute.assert_called_once_with( - "SELECT Name from CaseAttribute") + "SELECT Name from CaseAttribute" + ) @pytest.mark.unit_test def test_extract_invalid_csv_headers_with_some_wrong_headers(): """Test that invalid column headers are extracted correctly from a csv -string""" + string""" allowed_headers = [ - "Strain Name", "Value", "SE", "Count", - "Condition", "Tissue", "Sex", "Age", - "Ethn.", "PMI (hrs)", "pH", "Color", + "Strain Name", + "Value", + "SE", + "Count", + "Condition", + "Tissue", + "Sex", + "Age", + "Ethn.", + "PMI (hrs)", + "pH", + "Color", ] csv_text = "Strain Name, Value, SE, Colour" |