def test_model_validation(self):
"""
Tests the validation rules for the view.
"""
self.assertEqual(PartialView.objects.count(), 0)
PartialView(model=self.test_model, theta=0.0, phi=1.345).save()
self.assertEqual(PartialView.objects.count(), 1)
# If the set model, theta, phi is the same the view should not save
PartialView(model=self.test_model, theta=0.0, phi=1.345).save()
self.assertEqual(PartialView.objects.count(), 1)
PartialView(model=self.test_model, theta=1.0, phi=1.345).save()
self.assertEqual(PartialView.objects.count(), 2)
def setUp(self):
""" Run before starting testing. """
# TODO : use fixtures
test_model = SketchupModel()
test_model.google_id = "test1"
test_model.tags = ["tag1", "tag2"]
test_model.title = "title1"
test_model.text = "Description of 'title1' SketchupModel."
test_model.mesh = file("sketchup_models/fixtures/mesh_can.tri").read()
test_model.save()
self.test_model = SketchupModel.find_google_id("test1")
self.view = PartialView(model=self.test_model, theta=0.0, phi=0.0)
self.distribution = ShapeDistribution.compute(self.view.pointcloud)
def test_write_and_read(self):
"""
Tests writing then reading of a PointCloudStorage and ShapeDistribution.
"""
# Generate the pointcloud
view = PartialView(model=self.test_model, theta=0.0, phi=0.0)
self.assertTrue(view.pointcloud.size() > 0)
view.save()
self.assertEqual(PartialView.objects.count(), 1)
restored_view = PartialView.objects.get(pk=view.pk)
self.assertEqual(restored_view.theta, view.theta)
self.assertEqual(restored_view.phi, view.phi)
self.assertEqual(restored_view.model, view.model)
self.assertEqual(restored_view.pointcloud.size(),
view.pointcloud.size())
