Added test of Python binding to Spatial Pooler writeToString

htm-community · Jul 7, 2019 · 3c4d421 · 3c4d421
1 parent b59f646
commit 3c4d421
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/bindings/py/tests/algorithms/spatial_pooler_test.py b/bindings/py/tests/algorithms/spatial_pooler_test.py
@@ -177,6 +177,19 @@ def testNupicSpatialPoolerPickling(self):
     self.assertEqual(sp.getNumColumns(), sp2.getNumColumns(),
                      "Simple NuPIC SpatialPooler pickle/unpickle failed.")
 
+  def testNupicSpatialPoolerSavingToString(self):
+    """Test writing to and reading from NuPIC SpatialPooler."""
+
+    # Simple test: make sure that writing/reading works...
+    sp = SP()
+    s = sp.writeToString()
+
+    sp2 = sp.readFromString(s)
+
+    self.assertEqual(sp.getNumColumns(), sp2.getNumColumns(),
+                     "NuPIC SpatialPooler write to/read from string failed.")
+
+
 
 if __name__ == "__main__":
   unittest.main()