wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace "assert ...==..." with self.assertEqual

Browse Source
This commit is contained in:
Shinichi Hemmi
2023-07-27 17:17:21 +09:00
parent db0e1cc84c
commit 5b175e0d5b
5 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/artifact/test_boto3.py
+4 -4
View File
@@ -30,9 +30,9 @@ class Boto3BackendTestCase(TestCase):
backend = Boto3Backend(self.bucket_name)
backend.write(artifact_id, buf)
assert len(self.s3_client.list_objects(Bucket=self.bucket_name)["Contents"]) == 1
self.assertEqual(len(self.s3_client.list_objects(Bucket=self.bucket_name)["Contents"]), 1)
obj = self.s3_client.get_object(Bucket=self.bucket_name, Key=artifact_id)
assert obj["Body"].read() == dummy_content
self.assertEqual(obj["Body"].read(), dummy_content)
with backend.open(artifact_id) as f:
actual = f.read()
@@ -44,11 +44,11 @@ class Boto3BackendTestCase(TestCase):
backend = Boto3Backend(self.bucket_name)
backend.write(artifact_id, io.BytesIO(b"Hello"))
objects = self.s3_client.list_objects(Bucket=self.bucket_name)["Contents"]
assert len([obj for obj in objects if obj["Key"] == artifact_id]) == 1
self.assertEqual(len([obj for obj in objects if obj["Key"] == artifact_id]), 1)
backend.remove(artifact_id)
objects = self.s3_client.list_objects(Bucket=self.bucket_name).get("Contents", [])
assert len([obj for obj in objects if obj["Key"] == artifact_id]) == 0
self.assertEqual(len([obj for obj in objects if obj["Key"] == artifact_id]), 0)
def test_file_not_found_exception(self) -> None:
backend = Boto3Backend(self.bucket_name)