Added import export test for capsule layers

tests/unit/keras_app/test_views.py

@@ -697,6 +697,23 @@ def test_keras_import(self):
         self.pad_test(model, 'pad_w', 2)
 
 
+# **********Capsule Layers **************
+class CapsuleLayerTest(unittest.TestCase, HelperFunctions):
+    def setUp(self):
+        self.client = Client()
+
+    def test_keras_import_export(self):
+        model_file = open(os.path.join(settings.BASE_DIR, 'example/keras',
+                                       'Capsnet.json'), 'r')
+        response = self.client.post(reverse('keras-import'), {'file': model_file})
+        response = json.loads(response.content)
+        net = get_shapes(response['net'])
+        response = self.client.post(reverse('keras-export'), {'net': json.dumps(net),
+                                                              'net_name': ''})
+        response = json.loads(response.content)
+        self.assertEqual(response['result'], 'success')
+
+
 # ********** Export json tests **********
 
 # ********** Data Layers Test **********