def test_set_get_script(use_cluster):
    c = Client(None, use_cluster)
    sent_script = read_script_from_file()
    c.set_script("test-set-script", sent_script)
    script = c.get_script("test-set-script")
    assert sent_script == script
Example #2
0
def test_bad_device(use_cluster):
    c = Client(None, use_cluster)
    with pytest.raises(TypeError):
        c.set_script("key", "some_script", device="not-a-gpu")