nb_entity.py 文件源码

def test_cells(self, nb_test_case):
        # Execute all code cells in a notebook code entity, get the response message in JSON format,
        # and return all code cells parsed by messages in a list.
        errors = 0
        ws_url = nb_test_case.elyra_client.get_ws_kernel_endpoint(self.kernel_id)
        ws = websocket.create_connection(url=ws_url, timeout=itest_cell_timeout)
        print("Connection created for web socket {}".format(ws_url))
        try:
            code_cell_count = 1
            for code_cell in self.code_cell_list:
                if code_cell.is_executed():
                    errors = errors + NBCodeEntity.test_cell(nb_test_case, ws, code_cell, code_cell_count)
                code_cell_count = code_cell_count+1
        finally:
            ws.close()
        return errors