def test_stamp_without_revision(self, stamp): migration.stamp(None) stamp.assert_called_once_with('fake_config', 'head')
def test_stamp(self, stamp): migration.stamp('some_revision') stamp.assert_called_once_with('fake_config', 'some_revision')