compiler/rustc_index/src/vec/tests.rs

   1 #![allow(dead_code)]
   2
   3 // Allows the macro invocation below to work
   4 use crate as rustc_index;
   5
   6 rustc_macros::newtype_index!(struct MyIdx { MAX = 0xFFFF_FFFA });
   7
   8 #[test]
   9 fn index_size_is_optimized() {
  10     use std::mem::size_of;
  11
  12     assert_eq!(size_of::<MyIdx>(), 4);
  13     // Uses 0xFFFF_FFFB
  14     assert_eq!(size_of::<Option<MyIdx>>(), 4);
  15     // Uses 0xFFFF_FFFC
  16     assert_eq!(size_of::<Option<Option<MyIdx>>>(), 4);
  17     // Uses 0xFFFF_FFFD
  18     assert_eq!(size_of::<Option<Option<Option<MyIdx>>>>(), 4);
  19     // Uses 0xFFFF_FFFE
  20     assert_eq!(size_of::<Option<Option<Option<Option<MyIdx>>>>>(), 4);
  21     // Uses 0xFFFF_FFFF
  22     assert_eq!(size_of::<Option<Option<Option<Option<Option<MyIdx>>>>>>(), 4);
  23     // Uses a tag
  24     assert_eq!(size_of::<Option<Option<Option<Option<Option<Option<MyIdx>>>>>>>(), 8);
  25 }
  26
  27 #[test]
  28 fn range_iterator_iterates_forwards() {
  29     let range = MyIdx::from_u32(1)..MyIdx::from_u32(4);
  30     assert_eq!(
  31         range.collect::<Vec<_>>(),
  32         [MyIdx::from_u32(1), MyIdx::from_u32(2), MyIdx::from_u32(3)]
  33     );
  34 }
  35
  36 #[test]
  37 fn range_iterator_iterates_backwards() {
  38     let range = MyIdx::from_u32(1)..MyIdx::from_u32(4);
  39     assert_eq!(
  40         range.rev().collect::<Vec<_>>(),
  41         [MyIdx::from_u32(3), MyIdx::from_u32(2), MyIdx::from_u32(1)]
  42     );
  43 }
  44
  45 #[test]
  46 fn range_count_is_correct() {
  47     let range = MyIdx::from_u32(1)..MyIdx::from_u32(4);
  48     assert_eq!(range.count(), 3);
  49 }
  50
  51 #[test]
  52 fn range_size_hint_is_correct() {
  53     let range = MyIdx::from_u32(1)..MyIdx::from_u32(4);
  54     assert_eq!(range.size_hint(), (3, Some(3)));
  55 }