@@ -0,0 +1,1046 @@
+import unittest%0Aimport mock%0A%0Afrom wqflask import app%0A%0Afrom base.data_set import DatasetType%0A%0A    %0Aclass TestDataSetTypes(unittest.TestCase):%0A    def setUp(self):%0A        self.app_context = app.app_context()%0A        self.app_context.push()%0A%0A    def tearDown(self):%0A        self.app_context.pop()%0A%0A    @mock.patch('base.data_set.g')%0A    def test_data_set_type(self, db_mock):%0A        with app.app_context():%0A            db_mock.get = mock.Mock()%0A            r = mock.Mock()%0A            r.get.return_value = %22%22%22%0A            %7B%0A                %22AD-cases-controls-MyersGeno%22: %22Geno%22,%0A                %22AD-cases-controls-MyersPublish%22: %22Publish%22,%0A                %22AKXDGeno%22: %22Geno%22,%0A                %22AXBXAGeno%22: %22Geno%22,%0A                %22AXBXAPublish%22: %22Publish%22,%0A                %22Aging-Brain-UCIPublish%22: %22Publish%22,%0A                %22All Phenotypes%22: %22Publish%22,%0A                %22B139_K_1206_M%22: %22ProbeSet%22,%0A                %22B139_K_1206_R%22: %22ProbeSet%22%0A            %7D%0A            %22%22%22%0A            self.assertEqual(DatasetType(r)(%22All Phenotypes%22), %22Publish%22)%0A

Add unittests for *base/data_set*