diff --git a/migrate/tests/fixture/warnings.py b/migrate/tests/fixture/warnings.py index 2493708..346c6a2 100755 --- a/migrate/tests/fixture/warnings.py +++ b/migrate/tests/fixture/warnings.py @@ -13,7 +13,10 @@ class WarningMessage(object): local_values = locals() for attr in self._WARNING_DETAILS: setattr(self, attr, local_values[attr]) - self._category_name = category.__name__ if category else None + if category: + self._category_name = category.__name__ + else: + self._category_name = None def __str__(self): return ("{message : %r, category : %r, filename : %r, lineno : %s, "