Skip to content

Commit

Permalink
Add a simple unit test for get_energy
Browse files Browse the repository at this point in the history
  • Loading branch information
ajjackson committed May 28, 2024
1 parent 5543472 commit 57f7c83
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions tests/test_get_energy.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
import ase.build
from ase.calculators.singlepoint import SinglePointCalculator
import ase.io
import pytest

from mctools.generic.get_energy import get_energy


ENERGY = 3.141
FILENAME = "methane.extxyz"


@pytest.fixture
def methane_with_energy() -> ase.Atoms:
atoms = ase.build.molecule("CH4")
atoms.calc = SinglePointCalculator(atoms, energy=ENERGY)
return atoms


def test_get_energy(methane_with_energy, tmp_path) -> None:
ase.io.write(tmp_path / FILENAME, methane_with_energy)

assert get_energy(tmp_path / FILENAME) == pytest.approx(ENERGY)

0 comments on commit 57f7c83

Please sign in to comment.