aboutsummaryrefslogtreecommitdiff
path: root/tests/utils/common.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/utils/common.py')
-rw-r--r--tests/utils/common.py12
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