class TestViscousFriction(unittest.TestCase): """ Unit testing of ViscousFriction object. """ def setUp(self): """ Setup test subject. """ from pylith.friction.contrib.ViscousFriction import ViscousFriction self.model = ViscousFriction() return def test_label(self): """ Test constructor. """ label = "viscous friction" self.model.label(label) self.assertEqual(label, self.model.getLabel()) return def test_timeStep(self): """ Test constructor. """ dt = 2.4 self.model.timeStep(dt) self.assertAlmostEqual(dt, self.model.timeStep(), 5) return def test_factory(self): """ Test factory method. """ from pylith.friction.contrib.ViscousFriction import friction_model f = friction_model() return