def test_find_cite_petsc(self):
p = self.prob
p.setup()
p.model._vector_class.cite = "foobar PETScVector"
citations = find_citations(p)
# these two shouldn't have citations
self.assertFalse(IndepVarComp in citations)
self.assertFalse(NewtonSolver in citations)
try:
cite = citations[Problem]
self.assertEqual(p.cite, cite)
except KeyError:
self.fail('Citation for Problem class expected')
try:
cite = citations[Group]
self.assertEqual('foobar model', cite)
except KeyError:
self.fail('Citation for Group class expected')
try:
cite = citations[NonlinearRunOnce]
self.assertEqual('foobar nonlinear_solver', cite)
except KeyError:
self.fail('Citation for NonlinearRunOnce class expected')
try:
cite = citations[LinearRunOnce]
self.assertEqual('foobar linear_solver', cite)
except KeyError:
self.fail('Citation for LinearRunOnce class expected')
try:
cite = citations[ExecComp]
self.assertEqual('foobar exec comp', cite)
except KeyError:
self.fail('Citation for ExecComp class expected')
try:
cite = citations[PETScVector]
self.assertEqual('foobar PETScVector', cite)
except KeyError:
self.fail('Citation for PETScVector class expected')
def test_find_cite_petsc(self):
p = self.prob
p.setup()
p.model._vector_class.cite = "foobar PETScVector"
citations = find_citations(p)
# these two shouldn't have citations
self.assertFalse(IndepVarComp in citations)
self.assertFalse(NewtonSolver in citations)
try:
cite = citations[Problem]
self.assertEqual(p.cite, cite)
except KeyError:
self.fail('Citation for Problem class expected')
try:
cite = citations[Group]
self.assertEqual('foobar model', cite)
except KeyError:
self.fail('Citation for Group class expected')
try:
cite = citations[NonlinearRunOnce]
self.assertEqual('foobar nonlinear_solver', cite)
except KeyError:
self.fail('Citation for NonlinearRunOnce class expected')
try:
cite = citations[LinearRunOnce]
self.assertEqual('foobar linear_solver', cite)
except KeyError:
self.fail('Citation for LinearRunOnce class expected')
try:
cite = citations[ExecComp]
self.assertEqual('foobar exec comp', cite)
except KeyError:
self.fail('Citation for ExecComp class expected')
try:
cite = citations[PETScVector]
self.assertEqual('foobar PETScVector', cite)
except KeyError:
self.fail('Citation for PETScVector class expected')