From 1cd8e56c478b60b123fd7ba0106913cf3d64fec5 Mon Sep 17 00:00:00 2001
From: Matthew Tretter <m@tthewwithanm.com>
Date: Wed, 31 Jul 2013 22:19:41 -0400
Subject: [PATCH] Test ATX and ATX_CLOSED style headings

---
 tests/test_conversions.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/tests/test_conversions.py b/tests/test_conversions.py
index 75594bf..46a44ce 100644
--- a/tests/test_conversions.py
+++ b/tests/test_conversions.py
@@ -1,4 +1,4 @@
-from markdownify import markdownify as md
+from markdownify import markdownify as md, ATX, ATX_CLOSED
 
 
 def test_a():
@@ -54,6 +54,16 @@ def test_hn():
     assert md('<h6>Hello</h6>') == '###### Hello\n\n'
 
 
+def test_atx_headings():
+    assert md('<h1>Hello</h1>', heading_style=ATX) == '# Hello\n\n'
+    assert md('<h2>Hello</h2>', heading_style=ATX) == '## Hello\n\n'
+
+
+def test_atx_closed_headings():
+    assert md('<h1>Hello</h1>', heading_style=ATX_CLOSED) == '# Hello #\n\n'
+    assert md('<h2>Hello</h2>', heading_style=ATX_CLOSED) == '## Hello ##\n\n'
+
+
 def test_i():
     assert md('<i>Hello</i>') == '*Hello*'