RFD4-Proteina-prep / tests /test_prepare.py

Add the annotated-CIF preparation and export package for the Diffusers RFD4-Proteina pipeline

75b53c6 verified 22 days ago

541 Bytes

import importlib.util

import pytest


def test_missing_input_is_reported_before_any_work(tmp_path):
    pytest.importorskip("atomworks")
    import torch

    from rfd4_proteina_prep import prepare_cif

    with pytest.raises(FileNotFoundError):
        prepare_cif(tmp_path / "missing.cif", seed=0, device=torch.device("cpu"))


def test_public_api():
    import rfd4_proteina_prep

    assert set(rfd4_proteina_prep.__all__) == {"prepare_cif", "write_cif"}
    assert importlib.util.find_spec("rfd4_proteina_prep.processing") is not None

	import importlib.util

	import pytest


	def test_missing_input_is_reported_before_any_work(tmp_path):
	pytest.importorskip("atomworks")
	import torch

	from rfd4_proteina_prep import prepare_cif

	with pytest.raises(FileNotFoundError):
	prepare_cif(tmp_path / "missing.cif", seed=0, device=torch.device("cpu"))


	def test_public_api():
	import rfd4_proteina_prep

	assert set(rfd4_proteina_prep.__all__) == {"prepare_cif", "write_cif"}
	assert importlib.util.find_spec("rfd4_proteina_prep.processing") is not None