def test_particles_showing_cell_data(self):
        filename = self.filename
        cuds = create_example_particles()

        snapshot(cuds, filename, select=(CUBA.TEMPERATURE, 'bonds'))
        self.assertImageSavedWithContent(filename)

        with self.assertRaises(ValueError):
            snapshot(cuds, filename, select=(CUBA.TEMPERATURE, 'elements'))
    def test_lattice_showing_point_data(self):
        filename = self.filename
        cuds = create_example_lattice()

        def close(obj, event):
            obj.TerminateApp()

        snapshot(cuds, filename, select=(CUBA.TEMPERATURE, 'nodes'))
        self.assertImageSavedWithContent(filename)

        with self.assertRaises(ValueError):
            snapshot(cuds, filename, select=(CUBA.TEMPERATURE, 'particles'))

        with self.assertRaises(ValueError):
            snapshot(cuds, filename, select=(CUBA.TEMPERATURE, 'points'))
 def test_unknown_container(self):
     container = object()
     with self.assertRaises(TypeError):
         snapshot(container, self.filename)
 def test_cuds_snapshot(self, setup):
     cuds, _ = setup
     filename = self.filename
     snapshot(cuds, filename)
     self.assertImageSavedWithContent(filename)