[test] withContiguousStorage methods on raw buffer types

Author: glessard

validation-test/stdlib/UnsafeBufferPointer.swift.gyb

@@ -361,6 +361,31 @@ UnsafeMutableBufferPointerTestSuite.test("withContiguous(Mutable)StorageIfAvaila
   expectEqual(result1, result2)
 }
 
+UnsafeMutableRawBufferPointerTestSuite.test("withContiguous(Mutable)StorageIfAvailable") {
+  for c in [0, 1, 4] {
+    let pointer = UnsafeMutableRawPointer.allocate(byteCount: c, alignment: c)
+    defer { pointer.deallocate() }
+
+    let mutableBuffer = UnsafeMutableRawBufferPointer(start: pointer, count: c)
+    mutableBuffer.copyBytes(from: 0..<UInt8(c))
+
+    let r: ()? = mutableBuffer.withContiguousMutableStorageIfAvailable {
+      for i in $0.indices { $0[i] = 2*($0[i]+1) }
+    }
+    expectNotNil(r)
+    let s = mutableBuffer.withContiguousStorageIfAvailable {
+      $0.reduce(0, +)
+    }
+    expectEqual(s.map(Int.init), c*(c+1))
+
+    let immutableBuffer = UnsafeRawBufferPointer(mutableBuffer)
+    let t = immutableBuffer.withContiguousStorageIfAvailable {
+      $0.reduce(0, +)
+    }
+    expectEqual(s, t)
+  }
+}
+
 UnsafeMutableBufferPointerTestSuite.test("initialize(from: Slice)") {
   let o = 91
   let c = 1_000
@@ -403,6 +428,28 @@ UnsafeMutableBufferPointerTestSuite.test("Slice.withContiguousStorageIfAvailable
   }
 }
 
+UnsafeMutableRawBufferPointerTestSuite.test("Slice.withContiguousStorageIfAvailable") {
+  for test in subscriptRangeTests {
+    let c = test.collection.map({ UInt8($0.value/1000) })
+    let buffer = UnsafeMutableRawBufferPointer.allocate(
+      byteCount: test.collection.count, alignment: 8
+    )
+    defer { buffer.deallocate() }
+    buffer.copyBytes(from: c)
+
+    let expected = test.expected.map({ UInt8($0.value/1000) })
+    let r1: ()? = buffer[test.bounds].withContiguousStorageIfAvailable {
+      expectTrue(expected.elementsEqual($0))
+    }
+    expectNotNil(r1)
+    let r2: ()? = buffer[test.bounds].withContiguousMutableStorageIfAvailable {
+      for i in $0.indices { $0[i] += 1 }
+    }
+    expectNotNil(r2)
+    expectTrue(buffer[test.bounds].elementsEqual(expected.map({ $0+1 })))
+  }
+}
+
 UnsafeMutableBufferPointerTestSuite.test("sort") {
   var values = (0..<1000).map({ _ in Int.random(in: 0..<100) })
   let sortedValues = values.sorted()