fix a crash in each with 3+ arrays

src/algorithm/zip.rs

@@ -337,12 +337,10 @@ fn eachn(f: Function, mut args: Vec<Value>, env: &mut Uiua) -> UiuaResult {
  } else {
  let mut arg_elems: Vec<_> = args
  .into_iter()
- .map(|val| -> Box<dyn Iterator<Item = _>> {
- if val.rank() == 0 {
- Box::new(repeat(val).take(elem_count))
- } else {
- Box::new(val.into_elements())
- }
+ .map(|val| {
+ let repetitions = elem_count / val.element_count();
+ val.into_elements()
+ .flat_map(move |elem| repeat(elem).take(repetitions))
  })
  .collect();
  for _ in 0..elem_count {

src/value.rs

@@ -151,6 +151,17 @@ impl Value {
  Self::Box(array) => Box::new(array.into_rows().map(Value::from)),
  }
  }
+ /// Get an iterator over the elements of the value
+ pub fn elements(&self) -> Box<dyn ExactSizeIterator<Item = Self> + '_> {
+ match self {
+ Self::Num(array) => Box::new(array.data.iter().copied().map(Value::from)),
+ #[cfg(feature = "bytes")]
+ Self::Byte(array) => Box::new(array.data.iter().copied().map(Value::from)),
+ Self::Complex(array) => Box::new(array.data.iter().copied().map(Value::from)),
+ Self::Char(array) => Box::new(array.data.iter().copied().map(Value::from)),
+ Self::Box(array) => Box::new(array.data.iter().cloned().map(Value::from)),
+ }
+ }
  /// Cosume the value and get an iterator over its elements
  pub fn into_elements(self) -> Box<dyn Iterator<Item = Self>> {
  match self {

tests/loops.ua

@@ -86,6 +86,7 @@ A ← ↯2_3_4⇡24
 ⍤⟜≍: ∵(++) [1] [2 3 4] 5 ++ [1] [2 3 4] 5
 ⍤⟜≍: ⬚0∵(++) [1] [2 3 4] 5 ⬚0(++) [1] [2 3 4] 5
 ⍤⟜≍: ⬚0∵(++) [1] [2 3 4] [5 6] ⬚0(++) [1] [2 3 4] [5 6]
+⍤⟜≍: [[1_2_8 1_3_8 1_4_8] [1_5_9 1_6_9 1_7_9]] ∵[⊙⊙∘] 1 [2_3_4 5_6_7] 8_9
 
 ⍤⟜≍: [1_2 1_3 1_4] ≡⊟ 1 [2 3 4]
 ⍤⟜≍: [1_2 1_3 1_4] ⬚0≡⊟ 1 [2 3 4]