def test_errors_no_primitive_in_file(bad_primitives_files_dir): primitive_file = os.path.join(bad_primitives_files_dir, "no_primitives.py") error_text = 'No primitive defined in file %s' % primitive_file with pytest.raises(RuntimeError, match=error_text): load_primitive_from_file(primitive_file)
def test_errors_more_than_one_primitive_in_file(bad_primitives_files_dir): primitive_file = os.path.join(bad_primitives_files_dir, "multiple_primitives.py") error_text = 'More than one primitive defined in file %s' % primitive_file with pytest.raises(RuntimeError, match=error_text): load_primitive_from_file(primitive_file)
def test_load_primitive_from_file(primitives_to_install_dir): primitve_file = os.path.join(primitives_to_install_dir, "custom_max.py") primitive_name, primitive_obj = load_primitive_from_file(primitve_file) assert issubclass(primitive_obj, PrimitiveBase)