From 3174f1dab02537f4156fa2366855a36557a21200 Mon Sep 17 00:00:00 2001
From: Lizzy Fleckenstein <eliasfleckenstein@web.de>
Date: Mon, 6 Feb 2023 19:29:45 +0100
Subject: Support enumset

---
 lib/src/lib.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'lib/src')

diff --git a/lib/src/lib.rs b/lib/src/lib.rs
index d0e96e3..f68b9f7 100644
--- a/lib/src/lib.rs
+++ b/lib/src/lib.rs
@@ -128,6 +128,16 @@ where
     }
 }
 
+#[cfg(feature = "enumset")]
+impl<T: enumset::EnumSetType + GenerateRandom> GenerateRandom for enumset::EnumSet<T> {
+    fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
+        let max = enumset::EnumSet::<T>::variant_count() * 2;
+        let len = rng.gen_range(0..max);
+
+        (0..len).map(|_| T::generate_random(rng)).collect()
+    }
+}
+
 macro_rules! impl_generate_random_tuple {
 	( $t0:ident $( $t:ident )* ) => {
 		impl< $t0, $( $t, )* > GenerateRandom for ( $t0, $( $t, )* )
-- 
cgit v1.2.3