tests for the colnames header argument

2 files changed, 149 insertions, 1 deletions

diff --git a/testing/lisp/test-ob-R.el b/testing/lisp/test-ob-R.el
index bb9783f..2dfd141 100644
--- a/testing/lisp/test-ob-R.el
+++ b/testing/lisp/test-ob-R.el
@@ -24,7 +24,54 @@
       "#+begin_src R :session R\n  paste(\"Yep!\")\n#+end_src\n"
     (should (string= "Yep!" (org-babel-execute-src-block)))))
 
+(ert-deftest test-ob-R/colnames-yes-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames yes
+#+header: :var x = eg
+#+begin_src R
+x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") hline ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-R/colnames-nil-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames nil
+#+header: :var x = eg
+#+begin_src R
+x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") hline ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-R/colnames-no-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames no
+#+header: :var x = eg
+#+begin_src R
+x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
 (provide 'test-ob-R)
 
 ;;; test-ob-R.el ends here
- 
diff --git a/testing/lisp/test-ob-python.el b/testing/lisp/test-ob-python.el
new file mode 100644
index 0000000..e2990bc
--- /dev/null
+++ b/testing/lisp/test-ob-python.el
@@ -0,0 +1,101 @@
+;;; test-ob-python.el --- tests for ob-python.el
+
+;; Copyright (c) 2011 Eric Schulte
+;; Authors: Eric Schulte
+
+;; Released under the GNU General Public License version 3
+;; see: http://www.gnu.org/licenses/gpl-3.0.html
+
+(org-test-for-executable "python")
+
+(let ((load-path (cons (expand-file-name
+			".." (file-name-directory
+			      (or load-file-name buffer-file-name)))
+		       load-path)))
+  (require 'org-test)
+  (require 'org-test-ob-consts))
+
+(require 'ob-python)
+
+(ert-deftest test-ob-python/colnames-yes-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames yes
+#+header: :var x = eg
+#+begin_src python
+return x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") hline ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-python/colnames-yes-header-argument-again ()
+  (org-test-with-temp-text "#+tblname: less-cols
+| a |
+|---|
+| b |
+| c |
+
+#+header: :colnames yes
+#+begin_src python :var tab=less-cols
+  return [[val + '*' for val in row] for row in tab]
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("a") hline ("b*") ("c*"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-python/colnames-nil-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames nil
+#+header: :var x = eg
+#+begin_src python
+return x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") hline ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-python/colnames-no-header-argument-again ()
+  (org-test-with-temp-text "#+tblname: less-cols
+| a |
+|---|
+| b |
+| c |
+
+#+header: :colnames no
+#+begin_src python :var tab=less-cols
+  return [[val + '*' for val in row] for row in tab]
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("a*") ("b*") ("c*"))
+		   (org-babel-execute-src-block)))))
+
+(ert-deftest test-ob-python/colnames-no-header-argument ()
+  (org-test-with-temp-text "#+name: eg
+| col |
+|-----|
+| a   |
+| b   |
+
+#+header: :colnames no
+#+header: :var x = eg
+#+begin_src python
+return x
+#+end_src"
+    (org-babel-next-src-block)
+    (should (equal '(("col") ("a") ("b"))
+		   (org-babel-execute-src-block)))))
+
+(provide 'test-ob-python)
+
+;;; test-ob-python.el ends here
+