Skip to content
This repository has been archived by the owner on Jul 10, 2021. It is now read-only.

Commit

Permalink
Fixed last outstanding unit test post-serialisation.
Browse files Browse the repository at this point in the history
  • Loading branch information
alexjc committed Apr 23, 2015
1 parent 4343da5 commit d0c09f0
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions sknn/tests/test_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,7 @@ def test_PredictUninitialized(self):
a_in = numpy.zeros((8,16))
assert_raises(ValueError, self.nn.predict, a_in)

def __test_FitAutoInitialize(self):
# TODO: This hangs forever with serialization?
def test_FitAutoInitialize(self):
a_in, a_out = numpy.zeros((8,16)), numpy.zeros((8,4))
self.nn.fit(a_in, a_out)
assert_true(self.nn.is_initialized)
Expand Down Expand Up @@ -77,6 +76,11 @@ def setUp(self):
buf.seek(0)
self.nn = pickle.load(buf)

def test_FitAutoInitialize(self):
# Override base class test, you currently can't re-train a network that
# was serialized and deserialized.
pass

def test_PredictUninitialized(self):
# Override base class test, this is not initialized but it
# should be able to predict without throwing assert.
Expand Down

0 comments on commit d0c09f0

Please sign in to comment.