def test_compare_data_containers_not_equal(self):
expected = create_data_container()
data = create_data_container(constant=1)
with self.assertRaises(AssertionError):
compare_data_containers(data, expected, testcase=self)
data = create_data_container(restrict=[CUBA.MASS])
with self.assertRaises(AssertionError):
compare_data_containers(data, expected, testcase=self)
def test_compare_data_containers_not_equal(self):
expected = create_data_container()
data = create_data_container(constant=1)
with self.assertRaises(AssertionError):
compare_data_containers(data, expected, testcase=self)
data = create_data_container(restrict=[CUBA.MASS])
with self.assertRaises(AssertionError):
compare_data_containers(data, expected, testcase=self)
def test_create_points_with_id(self):
points = create_points_with_id(n=10)
self.assertEqual(len(points), 10)
for index, point in enumerate(points):
self.assertIsInstance(point, Point)
self.assertIsNotNone(point.uid)
self.assertEqual(point.coordinates,
(index, index * 10, index * 100))
compare_data_containers(point.data,
create_data_container(),
testcase=self)
def test_create_points_with_id(self):
points = create_points_with_id(n=10)
self.assertEqual(len(points), 10)
for index, point in enumerate(points):
self.assertIsInstance(point, Point)
self.assertIsNotNone(point.uid)
self.assertEqual(
point.coordinates, (index, index*10, index*100))
compare_data_containers(
point.data, create_data_container(),
testcase=self)
def test_create_particles_with_id(self):
particles = create_particles_with_id(n=9)
self.assertEqual(len(particles), 9)
for index, particle in enumerate(particles):
self.assertIsInstance(particle, Particle)
self.assertIsNotNone(particle.uid)
self.assertEqual(
particle.coordinates, (index, index*10, index*100))
compare_data_containers(
particle.data, create_data_container(),
testcase=self)
def test_create_particles_with_id(self):
particles = create_particles_with_id(n=9)
self.assertEqual(len(particles), 9)
for index, particle in enumerate(particles):
self.assertIsInstance(particle, Particle)
self.assertIsNotNone(particle.uid)
self.assertEqual(particle.coordinates,
(index, index * 10, index * 100))
compare_data_containers(particle.data,
create_data_container(),
testcase=self)
def vtk_compare_points(point, reference, msg=None, testcase=None):
''' use numpy.allclose to compare point coordinates retrieved
from vtk dataset with the reference as vtk casts coordinates to
double-precision floats and precision errors may be introduced
during casting
'''
self = testcase
self.assertEqual(point.uid, reference.uid)
if not numpy.allclose(point.coordinates, reference.coordinates):
error_message = "{} != {}"
self.failureException(error_message.format(point, reference))
compare_data_containers(point.data, reference.data, testcase=self)
def test_create_bonds(self):
n = 7
bonds = create_bonds(n=n)
self.assertEqual(len(bonds), n)
uids = set()
for index, bond in enumerate(bonds):
self.assertIsInstance(bond, Bond)
compare_data_containers(bond.data,
create_data_container(constant=index),
testcase=self)
self.assertEqual(len(bond.particles), n)
uids.update(bond.particles)
self.assertEqual(len(uids), n * n)
def test_create_bonds(self):
n = 7
bonds = create_bonds(n=n)
self.assertEqual(len(bonds), n)
uids = set()
for index, bond in enumerate(bonds):
self.assertIsInstance(bond, Bond)
compare_data_containers(
bond.data, create_data_container(constant=index),
testcase=self)
self.assertEqual(len(bond.particles), n)
uids.update(bond.particles)
self.assertEqual(len(uids), n*n)
def test_create_factory(self):
n_containers = 7
n_items = self.itemsPerContainer
containers = self.container_factory(n=n_containers, items=None)
self.assertEqual(len(containers), n_containers)
uids = set()
for index, container in enumerate(containers):
self.assertIsInstance(container, self.classtype)
compare_data_containers(
container.data, create_data_container(constant=index),
testcase=self)
self.assertEqual(len(container.points), n_items)
uids.update(container.points)
self.assertLessEqual(len(uids), n_containers*n_items)
def test_create_factory(self):
n_containers = 7
n_items = self.itemsPerContainer
containers = self.container_factory(n=n_containers, items=None)
self.assertEqual(len(containers), n_containers)
uids = set()
for index, container in enumerate(containers):
self.assertIsInstance(container, self.classtype)
compare_data_containers(container.data,
create_data_container(constant=index),
testcase=self)
self.assertEqual(len(container.points), n_items)
uids.update(container.points)
self.assertLessEqual(len(uids), n_containers * n_items)
def test_create_bonds_with_particles(self):
n = 7
particles = create_particles(n=100)
for particle in particles:
particle.uid = uuid.uuid4()
bonds = create_bonds(n=n, particles=particles)
uids = set()
self.assertEqual(len(bonds), n)
for index, bond in enumerate(bonds):
self.assertIsInstance(bond, Bond)
compare_data_containers(
bond.data, create_data_container(constant=index),
testcase=self)
self.assertEqual(len(bond.particles), n)
uids.update(bond.particles)
self.assertLessEqual(len(uids), n**2)
def test_create_bonds_with_particles(self):
n = 7
particles = create_particles(n=100)
for particle in particles:
particle.uid = uuid.uuid4()
bonds = create_bonds(n=n, particles=particles)
uids = set()
self.assertEqual(len(bonds), n)
for index, bond in enumerate(bonds):
self.assertIsInstance(bond, Bond)
compare_data_containers(bond.data,
create_data_container(constant=index),
testcase=self)
self.assertEqual(len(bond.particles), n)
uids.update(bond.particles)
self.assertLessEqual(len(uids), n**2)
def test_compare_data_containers_equal(self):
data = create_data_container()
expected = create_data_container()
# This should pass without a problem
compare_data_containers(data, expected, testcase=self)
compare_data_containers(
DataContainer(), DataContainer(), testcase=self)
compare_data_containers({}, DataContainer(), testcase=self)
def test_compare_data_containers_equal(self):
data = create_data_container()
expected = create_data_container()
# This should pass without a problem
compare_data_containers(data, expected, testcase=self)
compare_data_containers(DataContainer(),
DataContainer(),
testcase=self)
compare_data_containers({}, DataContainer(), testcase=self)
def compare_dummy_items(self, item, reference, msg=None):
self.assertEqual(item.uid, reference.uid)
self.assertEqual(item.value, reference.value)
compare_data_containers(item.data, reference.data, testcase=self)
def compare_dummy_items(self, item, reference, msg=None):
self.assertEqual(item.uid, reference.uid)
self.assertEqual(item.value, reference.value)
compare_data_containers(item.data, reference.data, testcase=self)
