Esempi in Python per LivyReliableHttpClient.get_statement

Esempio n. 1

File: test_livyreliablehttpclient.py Progetto: mindis/sparkmagic

def test_get_statement():
    http_client = MagicMock()
    livy_client = LivyReliableHttpClient(http_client)
    out = livy_client.get_statement(100, 4)
    assert_equals(out, http_client.get.return_value.json.return_value)
    http_client.get.assert_called_once_with("/sessions/100/statements/4",
                                            [200])

Esempio n. 2

File: test_livyreliablehttpclient.py Progetto: cfregly/sparkmagic

def test_get_statement():
    http_client = MagicMock()
    livy_client = LivyReliableHttpClient(http_client)
    out = livy_client.get_statement(100, 4)
    assert_equals(out, http_client.get.return_value.json.return_value)
    http_client.get.assert_called_once_with("/sessions/100/statements/4", [200])