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
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")