Python is_rule_signalled_gracetime_passed示例

示例#1

文件： test_utils.py 项目： awslabs/sagemaker-debugger

def test_is_rule_signalled_no_signal_file(monkeypatch):
    temp_file = tempfile.NamedTemporaryFile(mode="w+")
    time_str = str(int(time.time() - 2 * DEFAULT_GRACETIME_FOR_RULE_STOP_SEC))
    setup_rule_stop_file(temp_file, time_str, monkeypatch, write=False)
    dir = os.path.dirname(temp_file.name)
    # env variable is set, remove the file.
    temp_file.close()
    assert is_rule_signalled_gracetime_passed(dir) is False

示例#2

文件： test_utils.py 项目： awslabs/sagemaker-debugger

def test_is_rule_signalled_invalid_gracetime(monkeypatch):
    temp_file = tempfile.NamedTemporaryFile(mode="w+")
    setup_rule_stop_file(temp_file, "Invalid_time", monkeypatch)
    dir = os.path.dirname(temp_file.name)
    assert is_rule_signalled_gracetime_passed(dir) is True

示例#3

文件： test_utils.py 项目： awslabs/sagemaker-debugger

def test_is_rule_signalled_gracetime_passed(monkeypatch):
    temp_file = tempfile.NamedTemporaryFile(mode="w+")
    time_str = str(int(time.time() - 2 * DEFAULT_GRACETIME_FOR_RULE_STOP_SEC))
    setup_rule_stop_file(temp_file, time_str, monkeypatch)
    dir = os.path.dirname(temp_file.name)
    assert is_rule_signalled_gracetime_passed(dir) is True

示例#4

文件： test_utils.py 项目： awslabs/sagemaker-debugger

def test_is_rule_signalled_no_env_var_set(monkeypatch):
    assert is_rule_signalled_gracetime_passed("/fake-file") is False

示例#5

文件： test_utils.py 项目： awslabs/sagemaker-debugger

def test_is_rule_signalled_gracetime_not_passed(monkeypatch):
    temp_file = tempfile.NamedTemporaryFile(mode="w+")
    time_str = str(int(time.time()))
    setup_rule_stop_file(temp_file, time_str, monkeypatch)
    dir = os.path.dirname(temp_file.name)
    assert is_rule_signalled_gracetime_passed(dir) is False