def test_invalid_get_engine(self): """Raise error when forgetting to define an engine.""" with pytest.raises(NoEngineDefined) as excinfo: _get_engine() self.assertIn('registry, or not defined', str(excinfo.value))
def test_get_engine(self): """Get the defined engine.""" expected = 'bar' result = _get_engine() self.assertEqual(expected, result)