Fix flake8 compliance

BrainPad · Oct 28, 2024 · 9f4338e · 9f4338e
1 parent 28e23b7
commit 9f4338e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/cliboa/scenario/load/aws.py b/cliboa/scenario/load/aws.py
@@ -186,10 +186,12 @@ def _write_jsonl_to_dynamodb(self, file, table, partition_key, sort_key):
 
                     if partition_key not in item:
                         raise InvalidParameter(
-                            f"Partition key '{partition_key}' not found in JSONL file: {file} at line {line_number}"
+                            f"Partition key '{partition_key}' not found in JSONL file: "
+                            f"{file} at line {line_number}"
                         )
                     if sort_key and sort_key not in item:
                         raise InvalidParameter(
-                            f"Sort key '{sort_key}' not found in JSONL file: {file} at line {line_number}"
+                            f"Sort key '{sort_key}' not found in JSONL file: "
+                            f"{file} at line {line_number}"
                         )
                     batch.put_item(Item=item)
diff --git a/cliboa/test/scenario/load/test_aws.py b/cliboa/test/scenario/load/test_aws.py
@@ -12,10 +12,6 @@
 # all copies or substantial portions of the Software.
 #
 
-
-import csv
-import json
-import os
 import tempfile
 from unittest.mock import MagicMock, patch
 
@@ -98,7 +94,10 @@ def test_jsonl_success(self, mock_get_target_files, mock_boto3_resource):
         mock_boto3_resource.return_value.Table.return_value = mock_table
         mock_table.key_schema = [{"AttributeName": "id", "KeyType": "HASH"}]
 
-        jsonl_content = '{"id": "1", "name": "test1"}\n{"id": "2", "name": "test2"}'
+        jsonl_content = (
+            '{"id": "1", "name": "test1", "timestamp": "2023-01-01"}\n'
+            '{"id": "2", "name": "test2", "timestamp": "2023-01-02"}'
+        )
         with tempfile.NamedTemporaryFile(mode="w", delete=True) as temp_file:
             temp_file.write(jsonl_content)
             temp_file.flush()
@@ -187,7 +186,10 @@ def test_execute_jsonl_with_sort_key(self, mock_get_target_files, mock_boto3_res
             {"AttributeName": "timestamp", "KeyType": "RANGE"},
         ]
 
-        jsonl_content = '{"id": "1", "name": "test1", "timestamp": "2023-01-01"}\n{"id": "2", "name": "test2", "timestamp": "2023-01-02"}'
+        jsonl_content = (
+            '{"id": "1", "name": "test1", "timestamp": "2023-01-01"}\n'
+            '{"id": "2", "name": "test2", "timestamp": "2023-01-02"}'
+        )
         with tempfile.NamedTemporaryFile(mode="w", delete=True) as temp_file:
             temp_file.write(jsonl_content)
             temp_file.flush()