From 4b9956006ac5ffda3aae8faf2ad28c4f294a76e7 Mon Sep 17 00:00:00 2001
From: Sumpfork <github@petergorniak.org>
Date: Thu, 19 Nov 2015 17:20:08 -0800
Subject: [PATCH] first couple unit tests

---
 tests/carddb_tests.py | 33 +++++++++++++++++++++++++++++++++
 tests/layout_tests.py | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+)
 create mode 100644 tests/carddb_tests.py
 create mode 100644 tests/layout_tests.py

diff --git a/tests/carddb_tests.py b/tests/carddb_tests.py
new file mode 100644
index 0000000..1d3f42b
--- /dev/null
+++ b/tests/carddb_tests.py
@@ -0,0 +1,33 @@
+import unittest
+from .. import domdiv
+from ..domdiv import cards as domdiv_cards
+
+
+class TestCardDB(unittest.TestCase):
+
+    def test_cardread(self):
+        options, args = domdiv.parse_opts(['commandname'])
+        options.data_path = '.'
+        cards = domdiv.read_write_card_data(options)
+        self.assertEquals(len(cards), 312)
+        print set(c.cardset for c in cards)
+        valid_cardsets = {
+            u'prosperity',
+            u'cornucopia extras',
+            u'cornucopia',
+            u'promo',
+            u'adventures extras',
+            u'seaside',
+            u'adventures',
+            u'dark ages',
+            u'hinterlands',
+            u'dark ages extras',
+            u'alchemy',
+            u'base',
+            u'dominion',
+            u'guilds',
+            u'intrigue'
+        }
+        for c in cards:
+            self.assertIsInstance(c, domdiv_cards.Card)
+            self.assertIn(c.cardset, valid_cardsets)
diff --git a/tests/layout_tests.py b/tests/layout_tests.py
new file mode 100644
index 0000000..e643814
--- /dev/null
+++ b/tests/layout_tests.py
@@ -0,0 +1,35 @@
+import unittest
+from .. import domdiv
+from reportlab.lib.units import cm
+
+
+class TestLayout(unittest.TestCase):
+
+    def test_horizontal(self):
+        # should be the default
+        options, args = domdiv.parse_opts(['commandname'])
+        self.assertEquals(options.orientation, 'horizontal')
+        domdiv.calculate_layout(options)
+        self.assertEquals(options.numDividersHorizontal, 2)
+        self.assertEquals(options.numDividersVertical, 3)
+        self.assertEquals(options.dividerWidth, 9.1 * cm)
+        self.assertEquals(options.labelHeight, 0.9 * cm)
+        self.assertEquals(options.dividerHeight, 5.9 * cm + options.labelHeight)
+
+    def test_vertical(self):
+        options, args = domdiv.parse_opts(['commandname', '--orientation', 'vertical'])
+        self.assertEquals(options.orientation, 'vertical')
+        domdiv.calculate_layout(options)
+        self.assertEquals(options.numDividersHorizontal, 3)
+        self.assertEquals(options.numDividersVertical, 2)
+        self.assertEquals(options.dividerWidth, 5.9 * cm)
+        self.assertEquals(options.labelHeight, 0.9 * cm)
+        self.assertEquals(options.dividerHeight, 9.1 * cm + options.labelHeight)
+
+    def test_sleeved(self):
+        options, args = domdiv.parse_opts(['commandname', '--size', 'sleeved'])
+        domdiv.calculate_layout(options)
+        self.assertEquals(options.dividerWidth, 9.4 * cm)
+        self.assertEquals(options.labelHeight, 0.9 * cm)
+        self.assertEquals(options.dividerHeight, 6.15 * cm + options.labelHeight)
+