diff options
| -rw-r--r-- | libbe/storage/base.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/libbe/storage/base.py b/libbe/storage/base.py index 0f6b095..1b0ceba 100644 --- a/libbe/storage/base.py +++ b/libbe/storage/base.py @@ -960,8 +960,11 @@ if TESTING == True: storage_testcase_classes = [ c for c in ( ob for ob in globals().values() if isinstance(ob, type)) - if issubclass(c, StorageTestCase) \ - and c.Class == Storage] + if ((issubclass(c, StorageTestCase) \ + and c.Class == Storage) + or + (issubclass(c, VersionedStorageTestCase) \ + and c.Class == VersionedStorage))] for base_class in storage_testcase_classes: testcase_class_name = storage_class.__name__ + base_class.__name__ |