Esempio n. 1
0
class TestTraction(TestQuad4):
    """
  Test suite for testing pylith with 2-D axial tractions.
  """

    def setUp(self):
        """
    Setup for test.
    """
        TestQuad4.setUp(self)
        run_pylith()
        self.outputRoot = "lgdeformtraction"
        self.soln = AnalyticalSoln()
        return

    def calcDisplacements(self, vertices):
        """
    Calculate displacement field given coordinates of vertices.
    """
        return self.soln.displacement(vertices)

    def calcStateVar(self, name, vertices, cells):
        """
    Calculate state variable.
    """
        ncells = self.mesh["ncells"]
        pts = numpy.zeros((ncells, 2), dtype=numpy.float64)
        if name == "total_strain":
            stateVar = self.soln.strain(pts)
        elif name == "stress":
            stateVar = self.soln.stress(pts)
        else:
            raise ValueError("Unknown state variable '%s'." % name)

        return stateVar
Esempio n. 2
0
class TestTraction(TestQuad4):
  """
  Test suite for testing pylith with 2-D axial tractions.
  """

  def setUp(self):
    """
    Setup for test.
    """
    TestQuad4.setUp(self)
    run_pylith(TractionApp)
    self.outputRoot = "lgdeformtraction"
    self.soln = AnalyticalSoln()
    return


  def calcDisplacements(self, vertices):
    """
    Calculate displacement field given coordinates of vertices.
    """
    return self.soln.displacement(vertices)


  def calcStateVar(self, name, vertices, cells):
    """
    Calculate state variable.
    """
    ncells = self.mesh['ncells']
    pts = numpy.zeros( (ncells, 2), dtype=numpy.float64)
    if name == "total_strain":
      stateVar = self.soln.strain(pts)
    elif name == "stress":
      stateVar = self.soln.stress(pts)
    else:
      raise ValueError("Unknown state variable '%s'." % name)

    return stateVar