def test_convert_partially(): fe = fem.load_by_framework(framework="mock_py") assert fe is not None model = fe.load(path="") func = fe.convert_partially(model=model) stat = get_fe_stat(fe) assert stat.convert_partially == 1 fe.convert(function=func) stat = get_fe_stat(fe) assert stat.convert == 1
def test_decode_and_normalize(): fe = fem.load_by_framework(framework="mock_py") assert fe is not None model = fe.load(path="") func = fe.decode(model=model) stat = get_fe_stat(fe) assert stat.decode == 1 fe.normalize(function=func) stat = get_fe_stat(fe) assert stat.normalize == 1 assert stat.decode == 1
def test_load_by_model(): fe = fem.load_by_model(model_path="abc.test_mock_py_mdl") assert fe is not None assert fe.get_name() == "mock_py" stat = get_fe_stat(fe) assert stat.get_name == 1 assert stat.supported == 1
def test_load(): fe = fem.load_by_framework(framework="mock_py") assert fe is not None model = fe.load("abc.bin") assert model is not None stat = get_fe_stat(fe) assert "abc.bin" in stat.load_paths
def test_get_name(): fe = fem.load_by_framework(framework="mock_py") assert fe is not None name = fe.get_name() assert name == "mock_py" stat = get_fe_stat(fe) assert stat.get_name == 1
def test_convert_model(): fe = fem.load_by_framework(framework="mock_py") assert fe is not None model = fe.load(path="") func = fe.convert(model=model) assert func is not None stat = get_fe_stat(fe) assert stat.convert_model == 1
def test_load_by_framework_caps(): frontEnds = fem.get_available_front_ends() assert frontEnds is not None assert "mock_py" in frontEnds caps = [ FrontEndCapabilities.DEFAULT, FrontEndCapabilities.CUT, FrontEndCapabilities.NAMES, FrontEndCapabilities.WILDCARDS, FrontEndCapabilities.CUT | FrontEndCapabilities.NAMES | FrontEndCapabilities.WILDCARDS ] for cap in caps: fe = fem.load_by_framework(framework="mock_py", capabilities=cap) stat = get_fe_stat(fe) assert cap == stat.load_flags for i in range(len(caps) - 1): for j in range(i + 1, len(caps)): assert caps[i] != caps[j]