old_contents: """Tests for running lammps using CUDS and Simulation classes."""
  import unittest
  
  from simphony.core.cuba import CUBA
  from simphony import CUDS, Simulation
  from simphony.engine import EngineInterface
  from simphony.testing.utils import create_particles_with_id
  from simphony.cuds.particles import Particle, Particles
  
  
  class LAMMPSCUDSTestCase(unittest.TestCase):
      def setUp(self):
          self.cuds = self.generate_cuds()
  
      def generate_cuds(self):
          pset1 = create_particles_with_id(restrict=[CUBA.VELOCITY])
          pset2 = create_particles_with_id(restrict=[CUBA.VELOCITY])
  
          ps1 = Particles('ps1')
          ps2 = Particles('ps2')
  
          ps1.add_particles(pset1)
          ps2.add_particles(pset2)
  
          c = CUDS()
          c.add(ps1)
          c.add(ps2)
  
          return c
  
      def test_create_lammps_internal_simulation(self):
          self.assertRaisesRegexp(RuntimeError,
                                  'CUBAExtension.BOX_VECTORS',
                                  Simulation,
                                  self.cuds,
                                  'LAMMPS',
                                  EngineInterface.Internal)
  
      def test_create_lammps_fileio_simulation(self):
          sim = Simulation(self.cuds, 'LAMMPS', EngineInterface.FileIO)
  
      def test_create_liggghts_internal_simulation(self):
          self.assertRaisesRegexp(RuntimeError,
                                  'DEM using the INTERNAL interface is not yet supported',
                                  Simulation,
                                  self.cuds,
                                  'LIGGGHTS',
                                  EngineInterface.Internal)
  
      def test_create_liggghts_fileio_simulation(self):
          sim = Simulation(self.cuds, 'LIGGGHTS', EngineInterface.FileIO)
new_contents: """Tests for running lammps using CUDS and Simulation classes."""
  import unittest
  
  from simphony.core.cuba import CUBA
  from simphony import CUDS, Simulation
  from simphony.engine import EngineInterface
  from simphony.testing.utils import create_particles_with_id
  from simphony.cuds.particles import Particles
  
  
  class LAMMPSCUDSTestCase(unittest.TestCase):
      def setUp(self):
          self.cuds = self.generate_cuds()
  
      def generate_cuds(self):
          pset1 = create_particles_with_id(restrict=[CUBA.VELOCITY])
          pset2 = create_particles_with_id(restrict=[CUBA.VELOCITY])
  
          ps1 = Particles('ps1')
          ps2 = Particles('ps2')
  
          ps1.add_particles(pset1)
          ps2.add_particles(pset2)
  
          c = CUDS()
          c.add(ps1)
          c.add(ps2)
  
          return c
  
      def test_create_lammps_internal_simulation(self):
          self.assertRaisesRegexp(RuntimeError,
                                  'CUBAExtension.BOX_VECTORS',
                                  Simulation,
                                  self.cuds,
                                  'LAMMPS',
                                  EngineInterface.Internal)
  
      def test_create_lammps_fileio_simulation(self):
          sim = Simulation(self.cuds, 'LAMMPS', EngineInterface.FileIO)
          self.assertEqual(self.cuds, sim.get_cuds())
  
      def test_create_liggghts_internal_simulation(self):
          self.assertRaisesRegexp(RuntimeError,
                                  'DEM using the INTERNAL interface is not yet'
                                  ' supported',
                                  Simulation,
                                  self.cuds,
                                  'LIGGGHTS',
                                  EngineInterface.Internal)
  
      def test_create_liggghts_fileio_simulation(self):
          sim = Simulation(self.cuds, 'LIGGGHTS', EngineInterface.FileIO)
          self.assertEqual(self.cuds, sim.get_cuds())
  
subject: Fix flake 8