Fix builder tests

spookylukey · spookylukey · commit c8b831d469b3 · 2018-04-15T20:21:19.000+03:00
diff --git a/tests/test_builder.py b/tests/test_builder.py
@@ -1,28 +1,25 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 import unittest
-from creational.builder import construct_building, BuilderHouse,  BuilderFlat
+from creational.builder import construct_building, House, Flat, ComplexHouse
 
 
-class TestHouseBuilding(unittest.TestCase):
+class TestSimple(unittest.TestCase):
 
-    def setUp(self):
-        self.building = construct_building(BuilderHouse())
+    def test_house(self):
+        house = House()
+        self.assertEqual(house.size, 'Big')
+        self.assertEqual(house.floor, 'One')
 
-    def test_house_size(self):
-        self.assertEqual(self.building.size, 'Big')
+    def test_flat(self):
+        flat = Flat()
+        self.assertEqual(flat.size, 'Small')
+        self.assertEqual(flat.floor, 'More than One')
 
-    def test_num_floor_in_house(self):
-        self.assertEqual(self.building.floor, 'One')
 
+class TestComplex(unittest.TestCase):
 
-class TestFlatBuilding(unittest.TestCase):
-
-    def setUp(self):
-        self.building = construct_building(BuilderFlat())
-
-    def test_house_size(self):
-        self.assertEqual(self.building.size, 'Small')
-
-    def test_num_floor_in_house(self):
-        self.assertEqual(self.building.floor, 'More than One')
+    def test_house(self):
+        house = construct_building(ComplexHouse)
+        self.assertEqual(house.size, 'Big and fancy')
+        self.assertEqual(house.floor, 'One')
