Exemplo n.º 1
0
def test_set_stream():
    host = occa.host()
    stream = host.stream

    assert host == occa.get_device()

    stream2 = host.create_stream()

    assert stream != stream2

    occa.set_stream(stream2)

    assert stream != occa.get_stream()
Exemplo n.º 2
0
def test_get_stream():
    stream = occa.get_stream()

    assert isinstance(stream, occa.Stream)