Skip to content

Commit

Permalink
test: fix problem indexing tests
Browse files Browse the repository at this point in the history
  • Loading branch information
AayushSabharwal committed Jan 9, 2024
1 parent 5df156c commit bb01c9e
Showing 1 changed file with 5 additions and 11 deletions.
16 changes: 5 additions & 11 deletions test/downstream/problem_interface.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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)

Expand Down Expand Up @@ -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)
Expand All @@ -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)

Expand Down

0 comments on commit bb01c9e

Please sign in to comment.