def test_yamlIO(self): """Ensure we can read and write to YAML formatted streams.""" geom = SystemLayoutInput() geom.readGeomFromStream(io.StringIO(GEOM_INPUT)) fName = "testYamlIO.yaml" with open(fName, "w") as f: geom._writeYaml(f) # pylint: disable=protected-access with open(fName) as f: geom2 = SystemLayoutInput() geom2._readYaml(f) # pylint: disable=protected-access self.assertEqual(geom2.assemTypeByIndices[2, 2], "A2") os.remove(fName)
def test_growToFullCore(self): geom = SystemLayoutInput() geom.readGeomFromStream(io.StringIO(GEOM_INPUT)) self.assertNotIn((2, 3), geom.assemTypeByIndices) self.assertEqual(8, len(geom.assemTypeByIndices)) geom.growToFullCore() self.assertEqual(geometry.FULL_CORE, str(geom.symmetry.domain)) self.assertIn((2, 3), geom.assemTypeByIndices) self.assertIn( geom.assemTypeByIndices[2, 3], # perodic repeat geom.assemTypeByIndices[2, 1], ) # from input self.assertEqual(1 + 6 + 12, len(geom.assemTypeByIndices))
def test_asciimap(self): # pylint: disable=no-self-use """Ensure this can write ascii maps""" geom = SystemLayoutInput() geom.readGeomFromStream(io.StringIO(GEOM_INPUT)) geom._writeAsciiMap()