def get_nsfw_collections_srnames(): all_collections = Collection.get_all() nsfw_collections = [col for col in all_collections if col.over_18] srnames = itertools.chain.from_iterable(col.sr_names for col in nsfw_collections) return set(srnames)
def get_nsfw_collections_srnames(): all_collections = Collection.get_all() nsfw_collections = [col for col in all_collections if col.over_18] srnames = itertools.chain.from_iterable( col.sr_names for col in nsfw_collections ) return set(srnames)
def _assert_scenario(self, unordered_collections): CollectionStorage.get_all.return_value = unordered_collections self.assertEqual( [collection.name for collection in Collection.get_all()], self.correct_order)