Structure_test.cpp

Go to the documentation of this file.

#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>

#include "casm/crystallography/Structure.hh"

#include "casm/clex/PrimClex.hh"
#include "casm/app/AppIO.hh"
#include "casm/casm_io/VaspIO.hh"

using namespace CASM;

void prim1_read_test(Structure &struc) {

  double tol = 1e-5;

  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[0], Eigen::Vector3d(0.0, 2.0, 2.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[1], Eigen::Vector3d(2.0, 0.0, 2.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[2], Eigen::Vector3d(2.0, 2.0, 0.0), tol), true);
  BOOST_CHECK_EQUAL(struc.basis.size(), 1);

  // basis site 0 has three possible occupants
  BOOST_CHECK_EQUAL(struc.basis[0].site_occupant().size(), 3);

  std::string check_name[3] = {"A", "B", "C"};

  for(int i = 0; i < 3; i++) {
    // occupants are Molecule with name "A", etc.
    // Molecule are composed of AtomPosition
    // An AtomPosition 'is' a Coordinate with a Specie
    BOOST_CHECK_EQUAL(struc.basis[0].site_occupant()[i].name, check_name[i]);
    BOOST_CHECK_EQUAL(almost_equal(struc.basis[0].site_occupant()[i][0].const_frac(), Eigen::Vector3d(0.0, 0.0, 0.0), tol), true);
    BOOST_CHECK_EQUAL(struc.basis[0].site_occupant()[i][0].specie.name, check_name[i]);
  }

  // FCC motif
  BOOST_CHECK_EQUAL(48, struc.factor_group().size());
}

void prim2_read_test(Structure &struc) {

  double tol = 1e-5;

  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[0], Eigen::Vector3d(4.0, 0.0, 0.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[1], Eigen::Vector3d(0.0, 4.0, 0.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[2], Eigen::Vector3d(0.0, 0.0, 4.0), tol), true);
  BOOST_CHECK_EQUAL(struc.basis.size(), 4);

  // basis site 0 has three possible occupants
  BOOST_CHECK_EQUAL(struc.basis[0].site_occupant().size(), 3);

  std::string check_name[3] = {"A", "B", "C"};
  int check_value[4] = {0, 0, 1, 2};

  for(int i = 0; i < 4; i++) {
    for(int j = 0; j < 3; j++) {
      // occupants are Molecule with name "A", etc.
      // Molecule are composed of AtomPosition
      // An AtomPosition 'is' a Coordinate with a Specie
      BOOST_CHECK_EQUAL(struc.basis[i].site_occupant()[j].name, check_name[j]);
      BOOST_CHECK_EQUAL(almost_equal(struc.basis[i].site_occupant()[j][0].const_frac(), Eigen::Vector3d(0.0, 0.0, 0.0), tol), true);
      BOOST_CHECK_EQUAL(struc.basis[i].site_occupant()[j][0].specie.name, check_name[j]);

    }
    BOOST_CHECK_EQUAL(struc.basis[i].site_occupant().value(), check_value[i]);
  }

  // ordering on FCC motif
  BOOST_CHECK_EQUAL(16, struc.factor_group().size());
}


void pos1_read_test(Structure &struc) {

  double tol = 1e-5;

  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[0], Eigen::Vector3d(4.0, 0.0, 0.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[1], Eigen::Vector3d(0.0, 4.0, 0.0), tol), true);
  BOOST_CHECK_EQUAL(almost_equal(struc.lattice()[2], Eigen::Vector3d(0.0, 0.0, 4.0), tol), true);
  BOOST_CHECK_EQUAL(struc.basis.size(), 4);

  std::string check_name[4] = {"A", "A", "B", "C"};

  for(int i = 0; i < 4; i++) {
    // basis site 0 and 1 have one possible occupant
    BOOST_CHECK_EQUAL(struc.basis[i].site_occupant().size(), 1);

    // occupants are Molecule with name "A", etc.
    // Molecule are composed of AtomPosition
    // An AtomPosition 'is' a Coordinate with a Specie
    BOOST_CHECK_EQUAL(struc.basis[i].site_occupant()[0].name, check_name[i]);
    BOOST_CHECK_EQUAL(almost_equal(struc.basis[i].site_occupant()[0][0].const_frac(), Eigen::Vector3d(0.0, 0.0, 0.0), tol), true);
    BOOST_CHECK_EQUAL(struc.basis[i].site_occupant()[0][0].specie.name, check_name[i]);
  }

  // FCC structure
  BOOST_CHECK_EQUAL(16, struc.factor_group().size());
}

BOOST_AUTO_TEST_SUITE(StructureTest)

BOOST_AUTO_TEST_CASE(PRIM1Test) {

  fs::path testdir = "tests/unit/crystallography";

  // Read in test PRIM and run tests
  Structure struc(fs::path(testdir / "PRIM1"));
  prim1_read_test(struc);

  // Write test PRIM back out
  fs::path tmp_file = testdir / "PRIM1_out";
  write_prim(struc, tmp_file, FRAC);

  // Read new file and run tests again
  Structure struc2(read_prim(tmp_file));
  prim1_read_test(struc2);

}

BOOST_AUTO_TEST_CASE(PRIM2Test) {

  fs::path testdir = "tests/unit/crystallography";

  // Read in test PRIM and run tests
  Structure struc(fs::path(testdir / "PRIM2"));
  prim2_read_test(struc);
}

BOOST_AUTO_TEST_CASE(POS1Test) {

  fs::path testdir = "tests/unit/crystallography";

  // Read in test PRIM and run tests
  Structure struc(fs::path(testdir / "POS1"));
  pos1_read_test(struc);

  // Write test PRIM back out
  fs::path tmp_file = testdir / "POS1_out";
  fs::ofstream sout(tmp_file);
  VaspIO::PrintPOSCAR printer(struc);
  printer.set_append_atom_names_off();
  printer.print(sout);
  sout.close();

  // Read new file and run tests again
  Structure struc2(fs::path(testdir / "POS1_out"));
  pos1_read_test(struc2);

}

BOOST_AUTO_TEST_CASE(POS1Vasp5Test) {

  fs::path testdir = "tests/unit/crystallography";

  // Read in test PRIM and run tests
  Structure struc(fs::path(testdir / "POS1"));
  pos1_read_test(struc);

  // Write test PRIM back out
  fs::path tmp_file = testdir / "POS1_vasp5_out";
  fs::ofstream sout(tmp_file);
  VaspIO::PrintPOSCAR(struc).print(sout);
  sout.close();

  // Read new file and run tests again
  Structure struc2(fs::path(testdir / "POS1_vasp5_out"));
  pos1_read_test(struc2);

}

BOOST_AUTO_TEST_CASE(POS1jsonPrimTest) {

  fs::path testdir = "tests/unit/crystallography";

  // Read in test PRIM and run tests
  Structure struc(fs::path(testdir / "POS1"));
  pos1_read_test(struc);

  // Write test PRIM back out
  fs::path tmp_file = testdir / "POS1_prim.json";
  jsonParser json;
  write_prim(struc, json, FRAC);
  fs::ofstream sout(tmp_file);
  json.print(sout);
  sout.close();

  // Read new file and run tests again
  struc = Structure(read_prim(tmp_file));
  pos1_read_test(struc);

}

BOOST_AUTO_TEST_SUITE_END()