[stdlib] Test for the pointers difference

Signed-off-by: Manuel Saelices <msaelices@gmail.com>
msaelices · Sep 20, 2024 · 19830ec · 19830ec
1 parent cb72995
commit 19830ec
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/stdlib/test/memory/test_unsafepointer.mojo b/stdlib/test/memory/test_unsafepointer.mojo
@@ -282,6 +282,14 @@ def test_load_and_store_simd():
  assert_equal(ptr2[i], i // 4 * 4)
 
 
+def test_difference():
+ var ptr = UnsafePointer[Int].alloc(5)
+ var ptr2 = ptr + 2
+ assert_equal(ptr2 - ptr, 2)
+ assert_equal(ptr - ptr2, -2)
+ ptr.free()
+
+
 def main():
  test_address_of()
 
@@ -297,6 +305,7 @@ def main():
  test_unsafepointer_string()
  test_eq()
  test_comparisons()
+ test_difference()
 
  test_unsafepointer_address_space()
  test_indexing()