def test_lattice_hexagonal(self):
self.assertEqual(
main.Bravais2D(a=3.0, b=3.0, angle=60, plot=False).lattice,
'Hexagonal')
self.assertEqual(
main.Bravais2D(a=3.0, b=3.0, angle=120, plot=False).lattice,
'Hexagonal')
def test_numpoints_value(self):
self.assertEqual(
main.Bravais2D(numpoints=25, plot=False).numpoints, 25)
def test_lattice_square(self):
self.assertEqual(
main.Bravais2D(a=3.0, b=3.0, angle=90, plot=False).lattice,
'Square')
def test_angle_degrees(self):
self.assertEqual(
main.Bravais2D(angle=90, degrees=True, plot=False).angle,
math.pi / 2)
def test_lattice_rectangular(self):
self.assertEqual(
main.Bravais2D(a=2.0, b=1.0, angle=90, centered=False,
plot=False).lattice, 'Rectangular')
def test_lattice_oblique(self):
self.assertEqual(
main.Bravais2D(a=2.0, b=1.0, angle=45, plot=False).lattice,
'Oblique')
def test_b_vec(self):
bravais = main.Bravais2D(a=2.0, b=3.0, angle=90, plot=False)
self.assertEqual(bravais.b_vec[0],
bravais.a * bravais.b * math.cos(bravais.angle))
self.assertEqual(bravais.b_vec[1], bravais.b * math.sin(bravais.angle))
def test_a_vec(self):
bravais = main.Bravais2D(a=2.0, plot=False)
self.assertEqual(bravais.a_vec[0], bravais.a)
self.assertEqual(bravais.a_vec[1], 0)
def test_angle_radians(self):
self.assertEqual(
main.Bravais2D(angle=math.pi / 2, degrees=False, plot=False).angle,
math.pi / 2)