From 5093d72b9fece1a7c3d7a3d19e4161387ee6475a Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 25 Nov 2022 22:50:05 +0000
Subject: Use pytest instead of nose

---
 tests/transforms_tests.py | 55 +++++++++++++++++------------------------------
 1 file changed, 20 insertions(+), 35 deletions(-)

(limited to 'tests/transforms_tests.py')

diff --git a/tests/transforms_tests.py b/tests/transforms_tests.py
index d0604f8..94ce79a 100644
--- a/tests/transforms_tests.py
+++ b/tests/transforms_tests.py
@@ -1,61 +1,53 @@
 import cobble
-from nose.tools import assert_equal, istest
 
 from mammoth import documents, transforms
 from mammoth.transforms import get_descendants, get_descendants_of_type, _each_element
+from .testing import assert_equal
 
 
-@istest
 class ParagraphTests(object):
-    @istest
-    def paragraph_is_transformed(self):
+    def test_paragraph_is_transformed(self):
         paragraph = documents.paragraph(children=[])
         result = transforms.paragraph(lambda _: documents.tab())(paragraph)
         assert_equal(documents.tab(), result)
-        
-    @istest
-    def non_paragraph_elements_are_not_transformed(self):
+
+    def test_non_paragraph_elements_are_not_transformed(self):
         run = documents.run(children=[])
         result = transforms.paragraph(lambda _: documents.tab())(run)
         assert_equal(documents.run(children=[]), result)
 
 
-@istest
 class RunTests(object):
-    @istest
-    def run_is_transformed(self):
+    def test_run_is_transformed(self):
         run = documents.run(children=[])
         result = transforms.run(lambda _: documents.tab())(run)
         assert_equal(documents.tab(), result)
-        
-    @istest
-    def non_paragraph_elements_are_not_transformed(self):
+
+    def test_non_paragraph_elements_are_not_transformed(self):
         paragraph = documents.paragraph(children=[])
         result = transforms.run(lambda _: documents.tab())(paragraph)
         assert_equal(documents.paragraph(children=[]), result)
 
 
-@istest
 class EachElementTests(object):
-    @istest
-    def all_descendants_are_transformed(self):
+    def test_all_descendants_are_transformed(self):
         @cobble.data
         class Count(documents.HasChildren):
             count = cobble.field()
-        
+
         root = Count(count=None, children=[
             Count(count=None, children=[
                 Count(count=None, children=[]),
             ]),
         ])
-        
+
         current_count = [0]
         def set_count(node):
             current_count[0] += 1
             return node.copy(count=current_count[0])
-        
+
         result = _each_element(set_count)(root)
-        
+
         assert_equal(Count(count=3, children=[
             Count(count=2, children=[
                 Count(count=1, children=[]),
@@ -63,34 +55,27 @@ class EachElementTests(object):
         ]), result)
 
 
-@istest
 class GetDescendantsTests(object):
-    @istest
-    def returns_nothing_if_element_type_has_no_children(self):
+    def test_returns_nothing_if_element_type_has_no_children(self):
         assert_equal([], get_descendants(documents.tab()))
-    
-    @istest
-    def returns_nothing_if_element_has_empty_children(self):
+
+    def test_returns_nothing_if_element_has_empty_children(self):
         assert_equal([], get_descendants(documents.paragraph(children=[])))
-    
-    @istest
-    def includes_children(self):
+
+    def test_includes_children(self):
         children = [documents.text("child 1"), documents.text("child 2")]
         element = documents.paragraph(children=children)
         assert_equal(children, get_descendants(element))
-    
-    @istest
-    def includes_indirect_descendants(self):
+
+    def test_includes_indirect_descendants(self):
         grandchild = documents.text("grandchild")
         child = documents.run(children=[grandchild])
         element = documents.paragraph(children=[child])
         assert_equal([grandchild, child], get_descendants(element))
 
 
-@istest
 class GetDescendantsOfTypeTests(object):
-    @istest
-    def filters_descendants_to_type(self):
+    def test_filters_descendants_to_type(self):
         tab = documents.tab()
         run = documents.run(children=[])
         element = documents.paragraph(children=[tab, run])
-- 
cgit v1.2.3