예제 #1
0
def test_NCHW3():
    with pytest.raises(ValueError):
        check_input_format("rt")
예제 #2
0
def test_NCHW1():
    assert check_input_format("NCHW") == "NCHW"