Skip to content

Commit

Permalink
Update cuda_core/tests/test_stream.py
Browse files Browse the repository at this point in the history
Co-authored-by: Leo Fang <[email protected]>
  • Loading branch information
ksimpson-work and leofang authored Nov 4, 2024
1 parent 0c4d86f commit 925a42d
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions cuda_core/tests/test_stream.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,17 +71,13 @@ def test_stream_context():
context = stream.context
assert context is not None

def test_stream_device_with_foreign_stream():
def test_stream_from_foreign_stream():
device = Device()
other_stream = Stream._init(options=StreamOptions())
other_stream = device.create_stream(options=StreamOptions())
stream = device.create_stream(obj=other_stream)
assert other_stream.handle == stream.handle
device = stream.device
assert isinstance(device, Device)

def test_stream_context_with_foreign_stream():
device = Device()
other_stream = Stream._init(options=StreamOptions())
stream = device.create_stream(obj=other_stream)
context = stream.context
assert context is not None

Expand Down

0 comments on commit 925a42d

Please sign in to comment.