Updated unit test for keras exporter

Nirmal-Neel · Nirmal-Neel · commit 8129a37b9667 · 2019-02-27T13:09:41.000+05:30
diff --git a/nyoka/tests/keras_to_pmml_UnitTest.py b/nyoka/tests/keras_to_pmml_UnitTest.py
@@ -48,7 +48,7 @@ def test_keras_02(self):
         pmmlObj.export(open('sequentialModel.pmml','w'),0)
         reconPmmlObj=ny.parse('sequentialModel.pmml',True)
         self.assertEqual(os.path.isfile("sequentialModel.pmml"),True)
-        self.assertEqual(len(model.layers), len(reconPmmlObj.DeepNetwork[0].NetworkLayer))
+        self.assertEqual(len(model.layers), len(reconPmmlObj.DeepNetwork[0].NetworkLayer)-1)
 
 
 if __name__=='__main__':
