diff options
Diffstat (limited to 'tests/utils/common.py')
-rw-r--r-- | tests/utils/common.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/tests/utils/common.py b/tests/utils/common.py index 616a407..c29b47c 100644 --- a/tests/utils/common.py +++ b/tests/utils/common.py @@ -1,8 +1,10 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Common test utils module.""" from __future__ import annotations +from pathlib import Path + import numpy as np import tensorflow as tf @@ -30,3 +32,11 @@ def train_model(model: tf.keras.Model) -> None: x_train, y_train = get_dataset() model.fit(x_train, y_train, epochs=num_epochs) + + +def check_expected_permissions(path: Path, expected_permissions_mask: int) -> None: + """Check expected permissions for the provided path.""" + path_mode = path.stat().st_mode + permissions_mask = path_mode & 0o777 + + assert permissions_mask == expected_permissions_mask |