Update tests

Author: pxl-th

test/testsuite/alloc_cache.jl

@@ -54,7 +54,7 @@
         GPUArrays.@cached cache begin
             x5 = AT(zeros(T, dims))
 
-            # we're allowed to early free arrays, which shouldn't release the underlying data
+            # freeing arrays managed by cache is a no-op until they are released by cache.
             GPUArrays.unsafe_free!(x5)
         end
         _keys = collect(keys(cache.free))
@@ -63,11 +63,13 @@
         @test length(cache.free[key2]) == 1
         @test cache.free[key2][1] === GPUArrays.storage(x5)
 
-        # freeing all memory held by cache should free all allocations
         @test !GPUArrays.storage(x1).freed
-        @test GPUArrays.storage(x5).freed
-        @test GPUArrays.storage(x5).rc.count[] == 1 # the ref appears freed, but the data isn't
+        @test !GPUArrays.storage(x5).freed
+        # the refcount is not impacted by freeing until an array is not managed by cache
+        @test GPUArrays.storage(x5).rc.count[] == 2
         @test !GPUArrays.storage(x_free).freed
+
+        # freeing all memory held by cache should free all allocations
         GPUArrays.unsafe_free!(cache)
         @test sizeof(cache) == 0
         @test GPUArrays.storage(x1).freed