File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -311,10 +311,12 @@ abstract contract ERC4626Prop is Test {
311311 // s' = deposit(a)
312312 // s' <= s
313313 function prop_RT_withdraw_deposit (address caller , uint assets ) public {
314+ uint256 assetsBefore = vault_convertToAssets (IERC4626 (_vault_).balanceOf (caller)) + IERC20 (_underlying_).balanceOf (caller);
314315 vm.prank (caller); uint shares1 = vault_withdraw (assets, caller, caller);
315316 if (! _vaultMayBeEmpty) vm.assume (IERC20 (_vault_).totalSupply () > 0 );
316317 vm.prank (caller); uint shares2 = vault_deposit (assets, caller);
317- assertApproxLeAbs (shares2, shares1, _delta_);
318+ uint256 assetsAfter = vault_convertToAssets (IERC4626 (_vault_).balanceOf (caller)) + IERC20 (_underlying_).balanceOf (caller);
319+ assertApproxLeAbs (assetsAfter, assetsBefore, _delta_);
318320 }
319321
320322 //
You can’t perform that action at this time.
0 commit comments