Add tests for slice index methods

Author: mangelats

tests/index.rs

@@ -125,5 +125,124 @@ fn test_vec_ranges() {
     eq_its(soa.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
     unsafe { eq_its(soa.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
     eq_its(soa.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+}
+
+#[test]
+fn test_slice_usize() {
+    let mut aos = Vec::new();
+    let mut soa = ParticleVec::new();
+
+    let particle = Particle::new(String::from("Na"), 56.0);
+    aos.push(particle.clone());
+    soa.push(particle.clone());
+
+    // SoaIndex
+    let aos_slice = aos.as_slice();
+    let soa_slice = soa.as_slice();
+
+    assert_eq!(soa_slice.get(0).unwrap().name, &aos_slice.get(0).unwrap().name);
+    assert_eq!(soa_slice.get(0).unwrap().mass, &aos_slice.get(0).unwrap().mass);
+    assert_eq!(aos_slice.get(1), None);
+    assert_eq!(soa_slice.get(1), None);
+
+    unsafe {
+        assert_eq!(soa_slice.get_unchecked(0).name, &aos_slice.get_unchecked(0).name);
+        assert_eq!(soa_slice.get_unchecked(0).mass, &aos_slice.get_unchecked(0).mass);
+    }
+
+    assert_eq!(soa_slice.index(0).name, &aos_slice[0].name);
+    assert_eq!(soa_slice.index(0).mass, &aos_slice[0].mass);
+
+
+    // SoaIndexMut
+    let aos_mut_slice = aos.as_mut_slice();
+    let mut soa_mut_slice = soa.as_mut_slice();
+    assert_eq!(soa_mut_slice.get_mut(0).unwrap().name, &aos_mut_slice.get_mut(0).unwrap().name);
+    assert_eq!(soa_mut_slice.get_mut(0).unwrap().mass, &aos_mut_slice.get_mut(0).unwrap().mass);
+    assert_eq!(soa_mut_slice.get_mut(0).unwrap().mass, &aos_mut_slice.get_mut(0).unwrap().mass);
+    assert_eq!(aos_mut_slice.get_mut(1), None);
+    assert_eq!(soa_mut_slice.get_mut(1), None);
+
+    unsafe {
+        assert_eq!(soa_mut_slice.get_unchecked_mut(0).name, &aos_mut_slice.get_unchecked_mut(0).name);
+        assert_eq!(soa_mut_slice.get_unchecked_mut(0).mass, &aos_mut_slice.get_unchecked_mut(0).mass);
+    }
+
+    assert_eq!(soa_mut_slice.index_mut(0).name, &aos_mut_slice[0].name);
+    assert_eq!(soa_mut_slice.index_mut(0).mass, &aos_mut_slice[0].mass);
+
+
+    *soa_mut_slice.index_mut(0).mass -= 1.;
+    assert_eq!(soa_mut_slice.get(0).map(|p| *p.mass), Some(particle.mass - 1.));
+
+    *soa_mut_slice.get_mut(0).unwrap().mass += 1.;
+    assert_eq!(soa_mut_slice.get(0).map(|p| *p.mass), Some(particle.mass));
+}
+
+#[test]
+fn test_slice_ranges() {
+    let mut particles = Vec::new();
+    particles.push(Particle::new(String::from("Cl"), 1.0));
+    particles.push(Particle::new(String::from("Na"), 2.0));
+    particles.push(Particle::new(String::from("Br"), 3.0));
+    particles.push(Particle::new(String::from("Zn"), 4.0));
+
+    let mut soa = ParticleVec::new();
+
+    for particle in particles.iter() {
+        soa.push(particle.clone());
+    }
+
+    eq_its(soa.iter(), particles.iter());
 
-}
+    let mut soa_mut_slice = soa.as_mut_slice();
+    // All tests from here are the same only changing the range
+
+    let range = 0..1;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+
+    let range = ..3;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+
+    let range = 1..;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+
+    let range = ..;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+
+    let range = 0..=1;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+
+    let range = ..=2;
+    eq_its(soa_mut_slice.get(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index(range.clone()).iter(), particles[range.clone()].iter());
+    eq_its(soa_mut_slice.get_mut(range.clone()).unwrap().iter(), particles.get(range.clone()).unwrap().iter());
+    unsafe { eq_its(soa_mut_slice.get_unchecked_mut(range.clone()).iter(), particles.get_unchecked(range.clone()).iter()); }
+    eq_its(soa_mut_slice.index_mut(range.clone()).iter(), particles[range.clone()].iter());
+}