From cf585effcead8eb62cb3612eb8188dd6e20238a0 Mon Sep 17 00:00:00 2001 From: Kamil Gierszewski Date: Fri, 30 Aug 2024 15:38:02 +0200 Subject: [PATCH] test-framework: add generate random size method --- test_utils/size.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/test_utils/size.py b/test_utils/size.py index 8efcedf..851cc46 100644 --- a/test_utils/size.py +++ b/test_utils/size.py @@ -1,10 +1,12 @@ # # Copyright(c) 2019-2021 Intel Corporation +# Copyright(c) 2024 Huawei Technologies Co., Ltd. # SPDX-License-Identifier: BSD-3-Clause # import enum import math +import random from multimethod import multimethod @@ -209,3 +211,8 @@ class Size: @staticmethod def zero(): return Size(0) + + @staticmethod + def generate_random_size(min_size: int, max_size: int, unit: Unit): + size = random.randint(min_size, max_size) + return Size(value=float(size), unit=unit)