class MicrostructureTests(unittest.TestCase):
def setUp(self):
print('testing the Microstructure class')
self.test_eulers = [(45., 45, 0.), (10., 20, 30.), (191.9, 69.9, 138.9)]
self.micro = Microstructure()
self.micro.name = 'test'
for i in range(len(self.test_eulers)):
euler = self.test_eulers[i]
self.micro.grains.append(Grain(i + 1, Orientation.from_euler(euler)))
def test_to_h5(self):
self.micro.to_h5()
# read the file we have just written
m = Microstructure.from_h5('%s.h5' % self.micro.name, grain_centroid=None)
self.assertEqual(len(m.grains), len(self.test_eulers))
os.remove('%s.h5' % self.micro.name)
class MicrostructureTests(unittest.TestCase):
def setUp(self):
print('testing the Microstructure class')
self.test_eulers = [(45., 45, 0.), (10., 20, 30.),
(191.9, 69.9, 138.9)]
self.micro = Microstructure()
self.micro.name = 'test'
for i in range(len(self.test_eulers)):
euler = self.test_eulers[i]
self.micro.grains.append(
Grain(i + 1, Orientation.from_euler(euler)))
def test_to_h5(self):
self.micro.to_h5()
# read the file we have just written
m = Microstructure.from_h5('%s.h5' % self.micro.name)
self.assertEqual(len(m.grains), len(self.test_eulers))
os.remove('%s.h5' % self.micro.name)
def test_from_dct(self):
# read a microstructure from a DCT index.mat file
m = Microstructure.from_dct(data_dir=PYMICRO_EXAMPLES_DATA_DIR,
grain_file='t5_dct_cen_index.mat',
use_dct_path=False)
self.assertEqual(len(m.grains), 146)
self.assertEqual(m.voxel_size, 0.0014)
def test_from_h5(self):
# read a test microstructure
m = Microstructure.from_h5(
os.path.join(PYMICRO_EXAMPLES_DATA_DIR, 't5_dct_slice.h5'))
self.assertEqual(len(m.grains), 21)
self.assertEqual(m.voxel_size, 0.0014)
self.assertEqual(hasattr(m, 'grain_map'), True)
self.assertEqual(hasattr(m, 'mask'), True)
def test_find_neighbors(self):
m = Microstructure.from_h5(
os.path.join(PYMICRO_EXAMPLES_DATA_DIR, 't5_dct_slice.h5'))
neighbors = m.find_neighbors(grain_id=5, distance=3)
self.assertEqual(len(neighbors), 9)
for gid in [0, 1, 3, 14, 17, 18, 25, 51, 115]:
self.assertTrue(gid in neighbors)