From 00f3e4e0c6a2342b6fb1d0b041790ae596f93c4d Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@juliahub.com>
Date: Tue, 9 Jan 2024 17:48:39 +0530
Subject: [PATCH] test: fix problem indexing tests

---
 test/downstream/problem_interface.jl | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/test/downstream/problem_interface.jl b/test/downstream/problem_interface.jl
index 6757cf755..15f587d64 100644
--- a/test/downstream/problem_interface.jl
+++ b/test/downstream/problem_interface.jl
@@ -82,7 +82,7 @@ get_obs = getu(oprob, sys.x + sys.z + t + σ)
 @test gety(oprob) == 10.0
 @test get_arr(oprob) == [10.0, 10.0]
 @test get_tuple(oprob) == (10.0, 1.0)
-@test get_obs(oprob) == 39.0
+@test get_obs(oprob) == 22.0
 
 setx! = setu(oprob, x)
 sety! = setu(oprob, :y)
@@ -93,12 +93,8 @@ setx!(oprob, 11.0)
 @test getx(oprob) == 11.0
 sety!(oprob, 12.0)
 @test gety(oprob) == 12.0
-set_arr!(oprob, 10.0)
-@test get_arr(oprob) == [10.0, 10.0]
 set_arr!(oprob, [11.0, 12.0])
 @test get_arr(oprob) == [11.0, 12.0]
-set_tuple!(oprob, 13.0)
-@test get_tuple(oprob) == (13.0, 13.0)
 set_tuple!(oprob, [10.0, 10.0])
 @test get_tuple(oprob) == (10.0, 10.0)
 
@@ -144,12 +140,14 @@ getx = getu(sprob, x)
 gety = getu(sprob, :y)
 get_arr = getu(sprob, [x, y])
 get_tuple = getu(sprob, (y, z))
-get_obs = getu(sprob, sys.x + sys.z + t + σ)
+# observed doesn't work the same for SDEs
+# uncomment get_obs test below when fixed
+@test_broken get_obs = getu(sprob, sys.x + sys.z + t + σ)
 @test getx(sprob) == 10.0
 @test gety(sprob) == 10.0
 @test get_arr(sprob) == [10.0, 10.0]
 @test get_tuple(sprob) == (10.0, 1.0)
-@test get_obs(sprob) == 39.0
+# @test get_obs(sprob) == 39.0
 
 setx! = setu(sprob, x)
 sety! = setu(sprob, :y)
@@ -160,12 +158,8 @@ setx!(sprob, 11.0)
 @test getx(sprob) == 11.0
 sety!(sprob, 12.0)
 @test gety(sprob) == 12.0
-set_arr!(sprob, 10.0)
-@test get_arr(sprob) == [10.0, 10.0]
 set_arr!(sprob, [11.0, 12.0])
 @test get_arr(sprob) == [11.0, 12.0]
-set_tuple!(sprob, 13.0)
-@test get_tuple(sprob) == (13.0, 13.0)
 set_tuple!(sprob, [10.0, 10.0])
 @test get_tuple(sprob) == (10.0, 10.0)