823173a761
This PR fixes `ST.Ref.ptrEq` to act as described in the docs. This fixes two bugs: 1. The recent `IO.RealWorld` elimination PR overlooked this function (afaik this is the only one), causing its return value to be generally wrong. 2. The implementation of `ptrEq` would previously always consider two different cells with pointer equivalent value to be pointer equal. However, the function is supposed to check whether two `Ref` are the same cell, not whether the contained elements are.
85 lines
1.2 KiB
Text
85 lines
1.2 KiB
Text
/-!
|
|
Some basic tests for the ST monad.
|
|
-/
|
|
|
|
namespace STTest
|
|
|
|
def ptrEq : IO Unit := do
|
|
let ref1 ← IO.mkRef 0
|
|
let ref2 ← IO.mkRef 0
|
|
IO.println (← ref1.ptrEq ref1)
|
|
IO.println (← ref1.ptrEq ref2)
|
|
|
|
/--
|
|
info: true
|
|
false
|
|
-/
|
|
#guard_msgs in
|
|
#eval ptrEq
|
|
|
|
def readWriteRegister : IO Unit := do
|
|
let ref1 ← IO.mkRef 0
|
|
IO.println (← ref1.get)
|
|
ref1.set 1
|
|
IO.println (← ref1.get)
|
|
|
|
/--
|
|
info: 0
|
|
1
|
|
-/
|
|
#guard_msgs in
|
|
#eval readWriteRegister
|
|
|
|
def swapRegister : IO Unit := do
|
|
let ref1 ← IO.mkRef 0
|
|
IO.println (← ref1.swap 5)
|
|
IO.println (← ref1.get)
|
|
|
|
/--
|
|
info: 0
|
|
5
|
|
-/
|
|
#guard_msgs in
|
|
#eval swapRegister
|
|
|
|
unsafe def takeRegister : IO Unit := do
|
|
let ref1 ← IO.mkRef 0
|
|
IO.println (← ref1.take)
|
|
ref1.set 5
|
|
IO.println (← ref1.get)
|
|
|
|
/--
|
|
info: 0
|
|
5
|
|
-/
|
|
#guard_msgs in
|
|
#eval takeRegister
|
|
|
|
def modifyRegister : IO Unit := do
|
|
let ref1 ← IO.mkRef 1
|
|
IO.println (← ref1.get)
|
|
ref1.modify (fun x => 2 * x)
|
|
IO.println (← ref1.get)
|
|
|
|
/--
|
|
info: 1
|
|
2
|
|
-/
|
|
#guard_msgs in
|
|
#eval modifyRegister
|
|
|
|
def modifyGetRegister : IO Unit := do
|
|
let ref1 ← IO.mkRef 1
|
|
IO.println (← ref1.get)
|
|
IO.println (← ref1.modifyGet (fun x => (x, 2 * x)))
|
|
IO.println (← ref1.get)
|
|
|
|
/--
|
|
info: 1
|
|
1
|
|
2
|
|
-/
|
|
#guard_msgs in
|
|
#eval modifyGetRegister
|
|
|
|
end STTest
|