mirror of
https://github.com/vale981/emacs-ipython-notebook
synced 2025-03-06 01:21:38 -05:00
122 lines
4.6 KiB
EmacsLisp
122 lines
4.6 KiB
EmacsLisp
;;; ein-testing-notebook.el --- Testing utilities for notebook module
|
|
|
|
;; Copyright (C) 2012 Takafumi Arakaki
|
|
|
|
;; Author: Takafumi Arakaki <aka.tkf at gmail.com>
|
|
|
|
;; This file is NOT part of GNU Emacs.
|
|
|
|
;; ein-testing-notebook.el is free software: you can redistribute it
|
|
;; and/or modify it under the terms of the GNU General Public License
|
|
;; as published by the Free Software Foundation, either version 3 of
|
|
;; the License, or (at your option) any later version.
|
|
|
|
;; ein-testing-notebook.el is distributed in the hope that it will be useful,
|
|
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
;; GNU General Public License for more details.
|
|
|
|
;; You should have received a copy of the GNU General Public License
|
|
;; along with ein-testing-notebook.el.
|
|
;; If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
;;; Commentary:
|
|
|
|
;;
|
|
|
|
;;; Code:
|
|
|
|
(eval-when-compile (require 'cl))
|
|
|
|
(require 'ein-notebook)
|
|
(require 'ein-testing-cell)
|
|
|
|
(defun ein:testing-notebook-from-json (json-string &optional name path)
|
|
(let* ((data (ein:json-read-from-string json-string))
|
|
(name (plist-get data :name))
|
|
(path (plist-get data :path))
|
|
(kernelspec (make-ein:$kernelspec :name "python3"))
|
|
(content (make-ein:$content :url-or-port "DUMMY-URL"
|
|
:ipython-version 3
|
|
:path path)))
|
|
(unless name (setq name "NOTEBOOK-DUMMY"))
|
|
(unless path (setq path "NOTEBOOK-DUMMY"))
|
|
;; cl-flet does not work correctly here!
|
|
(cl-flet ((pop-to-buffer (buf)
|
|
buf)
|
|
(ein:query-ipython-version (&optional url-or-port force)
|
|
3)
|
|
(ein:notebook-start-kernel (notebook)
|
|
notebook))
|
|
(let ((notebook (ein:notebook-new "DUMMY-URL" path kernelspec)))
|
|
(setf (ein:$notebook-kernel notebook)
|
|
(ein:kernel-new 8888 "/kernels" (ein:$notebook-events notebook) (ein:query-ipython-version)))
|
|
(setf (ein:$kernel-events (ein:$notebook-kernel notebook))
|
|
(ein:events-new))
|
|
(ein:notebook-request-open-callback notebook (ein:new-content content nil :data data))
|
|
(ein:notebook-buffer notebook)))))
|
|
|
|
(defun ein:testing-notebook-make-data (cells &optional name path)
|
|
(setq cells
|
|
(ein:testing-notebook--preprocess-cells-data-for-json-encode cells))
|
|
(unless name (setq name "Dummy Name.ipynb"))
|
|
(unless path (setq path "Dummy Name.ipynb"))
|
|
`((path . ,path)
|
|
(name . ,name)
|
|
(type . "notebook")
|
|
(format . "json")
|
|
(mimetype . nil)
|
|
(writeable . t)
|
|
(content (metadata . ())
|
|
(nbformat . 4)
|
|
(nbformat_minor . 0)
|
|
(cells . ,(apply #'vector cells)))))
|
|
|
|
(defun ein:testing-notebook--preprocess-cells-data-for-json-encode (cells)
|
|
"Preprocess CELLS data to make it work nice with `json-encode'."
|
|
(mapcar (lambda (c)
|
|
(cond
|
|
((equal (plist-get c :cell_type) "code")
|
|
;; turn `:outputs' into an array.
|
|
(plist-put c :outputs (apply #'vector (plist-get c :outputs))))
|
|
(t c)))
|
|
cells))
|
|
|
|
(defun ein:testing-notebook-make-new (&optional name cells-data)
|
|
"Make new notebook. One empty cell will be inserted
|
|
automatically if CELLS-DATA is nil."
|
|
(ein:testing-notebook-from-json
|
|
(json-encode (ein:testing-notebook-make-data cells-data name))))
|
|
|
|
(defun ein:testing-notebook-make-empty (&optional name)
|
|
"Make empty notebook and return its buffer.
|
|
Automatically inserted cell for new notebook is deleted."
|
|
(let ((buffer (ein:testing-notebook-make-new name)))
|
|
(with-current-buffer buffer
|
|
(call-interactively #'ein:worksheet-delete-cell))
|
|
buffer))
|
|
|
|
(defmacro ein:testing-with-one-cell (cell-type &rest body)
|
|
"Insert new cell of CELL-TYPE in a clean notebook and execute BODY.
|
|
The new cell is bound to a variable `cell'."
|
|
(declare (indent 1))
|
|
`(with-current-buffer (ein:testing-notebook-make-empty)
|
|
(let ((cell (ein:worksheet-insert-cell-below ein:%worksheet%
|
|
,cell-type nil t)))
|
|
,@body)))
|
|
|
|
(defun ein:testing-make-notebook-with-outputs (list-outputs)
|
|
"Make a new notebook with cells with output.
|
|
LIST-OUTPUTS is a list of list of strings (pyout text). Number
|
|
of LIST-OUTPUTS equals to the number cells to be contained in the
|
|
notebook."
|
|
(ein:testing-notebook-make-new
|
|
nil
|
|
(mapcar (lambda (outputs)
|
|
(ein:testing-codecell-data
|
|
nil nil (mapcar #'ein:testing-codecell-pyout-data outputs)))
|
|
list-outputs)))
|
|
|
|
(provide 'ein-testing-notebook)
|
|
|
|
;;; ein-testing-notebook.el ends here
|