From 519b892f0c51097b8b43eef070d33d05ec4deee1 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Tue, 3 Oct 2023 14:24:52 -0400 Subject: [PATCH] Add unit test for Struct_mallinfo2.__str__() --- invokeai/backend/model_management/libc_util.py | 1 + tests/backend/model_management/test_libc_util.py | 9 ++++++++- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/model_management/libc_util.py b/invokeai/backend/model_management/libc_util.py index 68074235fa..8a0f4875aa 100644 --- a/invokeai/backend/model_management/libc_util.py +++ b/invokeai/backend/model_management/libc_util.py @@ -50,6 +50,7 @@ class Struct_mallinfo2(ctypes.Structure): ) s += f"{'fordblks': <10}= {(self.fordblks/2**30):15.5f} # Space in free blocks (non-mmapped) (GB)\n" s += f"{'keepcost': <10}= {(self.keepcost/2**30):15.5f} # Top-most, releasable space (GB)\n" + return s class LibcUtil: diff --git a/tests/backend/model_management/test_libc_util.py b/tests/backend/model_management/test_libc_util.py index 54af88515d..eda12036b0 100644 --- a/tests/backend/model_management/test_libc_util.py +++ b/tests/backend/model_management/test_libc_util.py @@ -1,6 +1,6 @@ import pytest -from invokeai.backend.model_management.libc_util import LibcUtil +from invokeai.backend.model_management.libc_util import LibcUtil, Struct_mallinfo2 def test_libc_util_mallinfo2(): @@ -14,3 +14,10 @@ def test_libc_util_mallinfo2(): info = libc.mallinfo2() assert info.arena > 0 + + +def test_struct_mallinfo2_to_str(): + """Smoke test of Struct_mallinfo2.__str__().""" + info = Struct_mallinfo2() + info_str = str(info) + print(info_str)