diff options
| -rw-r--r-- | gn3/db/sample_data.py | 8 | ||||
| -rw-r--r-- | tests/unit/db/test_sample_data.py | 4 |
2 files changed, 6 insertions, 6 deletions
diff --git a/gn3/db/sample_data.py b/gn3/db/sample_data.py index 6482116..e5c5e48 100644 --- a/gn3/db/sample_data.py +++ b/gn3/db/sample_data.py @@ -1,5 +1,5 @@ from gn3.csvcmp import extract_strain_name -from typing import Any, Tuple, Union +from typing import Any, Tuple import MySQLdb @@ -333,9 +333,9 @@ def insert_sample_data(conn: Any, # pylint: disable=[R0913] # Check if the data already exists: with conn.cursor() as cursor: cursor.execute( - "SELECT Id FROM PublishData where Id = %s " - "AND StrainId = %s", - (data_id, strain_id)) + "SELECT Id FROM PublishData where Id = %s " + "AND StrainId = %s", + (data_id, strain_id)) if cursor.fetchone(): # Data already exists return count diff --git a/tests/unit/db/test_sample_data.py b/tests/unit/db/test_sample_data.py index 80166fc..84ab1a1 100644 --- a/tests/unit/db/test_sample_data.py +++ b/tests/unit/db/test_sample_data.py @@ -87,7 +87,7 @@ def test_extract_actions(): "delete": {"data": "18", "csv_header": "Value"}, "insert": {"data": "2,F", "csv_header": "SE,Sex"}, "update": {"data": "1", "csv_header": "Count"}, - }) + }) assert(__extract_actions(original_data="BXD1,18,x,0,x", updated_data="BXD1,19,2,1,F", csv_header="Strain Name,Value,SE,Count,Sex") == @@ -95,7 +95,7 @@ def test_extract_actions(): "delete": None, "insert": {"data": "2,F", "csv_header": "SE,Sex"}, "update": {"data": "19,1", "csv_header": "Value,Count"}, - }) + }) @pytest.mark.unit_test |