From 4c86400d3276b90eb2b92a244936ef57d7cfe3a1 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Tue, 8 May 2018 17:22:42 -0700
Subject: [PATCH] cleanup deck tests

---
 tests/test_decks.py | 35 ++++++++++++++---------------------
 1 file changed, 14 insertions(+), 21 deletions(-)

diff --git a/tests/test_decks.py b/tests/test_decks.py
index f71601d..58c3b9c 100755
--- a/tests/test_decks.py
+++ b/tests/test_decks.py
@@ -4,37 +4,30 @@ import unittest
 import util
 
 
-class TestDeckNames(unittest.TestCase):
+class TestDecks(unittest.TestCase):
     def runTest(self):
-        result = util.invoke('deckNames')
-        self.assertIn('Default', result)
+        # deckNames (part 1)
+        deckNames = util.invoke('deckNames')
+        self.assertIn('Default', deckNames)
 
-
-class TestDeckNamesAndIds(unittest.TestCase):
-    def runTest(self):
+        # deckNamesAndIds
         result = util.invoke('deckNamesAndIds')
         self.assertIn('Default', result)
         self.assertEqual(result['Default'], 1)
 
-
-class TestCreateDeck(unittest.TestCase):
-    def tearDown(self):
-        util.invoke('deleteDecks', decks=['test'])
-
-
-    def runTest(self):
-        util.invoke('createDeck', deck='test')
-        self.assertIn('test', util.invoke('deckNames'))
-
-
-class TestDeleteDecks(unittest.TestCase):
-    def setUp(self):
+        # createDeck
         util.invoke('createDeck', deck='test')
 
+        # deckNames (part 2)
+        deckNames = util.invoke('deckNames')
+        self.assertIn('test', deckNames)
 
-    def runTest(self):
+        # deleteDecks
         util.invoke('deleteDecks', decks=['test'])
-        self.assertNotIn('test', util.invoke('deckNames'))
+
+        # deckNames (part 3)
+        deckNames = util.invoke('deckNames')
+        self.assertNotIn('test', deckNames)
 
 
 if __name__ == '__main__':