class TestServer(object): def setUp(self): self.server_process = Process( target=start_server, args=(get_stream(),)) self.server_process.start() self.stream = ServerDataStream(('f', 't'), False) def tearDown(self): self.server_process.terminate() self.stream = None def test_server(self): server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): assert_allclose(*data) assert_raises(StopIteration, next, server_data) def test_pickling(self): try: self.stream = cPickle.loads(cPickle.dumps(self.stream)) # regression test: pickling of an unpickled stream used it fail cPickle.dumps(self.stream) server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): assert_allclose(*data, rtol=1e-3) except AssertionError as e: raise SkipTest("Skip test_that failed with: {}".format(e)) assert_raises(StopIteration, next, server_data) def test_value_error_on_request(self): assert_raises(ValueError, self.stream.get_data, [0, 1]) def test_close(self): self.stream.close() def test_next_epoch(self): self.stream.next_epoch() def test_reset(self): self.stream.reset()
class TestServer(object): def setUp(self): self.server_process = Process( target=start_server, args=(get_stream(),)) self.server_process.start() self.stream = ServerDataStream(('f', 't'), False) def tearDown(self): self.server_process.terminate() self.stream = None def test_server(self): server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): assert_allclose(*data) assert_raises(StopIteration, next, server_data) def test_pickling(self): self.stream = cPickle.loads(cPickle.dumps(self.stream)) server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): #pass assert_allclose(*data, rtol=1e-5) assert_raises(StopIteration, next, server_data) def test_value_error_on_request(self): assert_raises(ValueError, self.stream.get_data, [0, 1]) def test_close(self): self.stream.close() def test_next_epoch(self): self.stream.next_epoch() def test_reset(self): self.stream.reset()
class TestServer(object): def setUp(self): self.server_process = Process(target=start_server, args=(get_stream(), )) self.server_process.start() self.stream = ServerDataStream(('f', 't'), False) def tearDown(self): self.server_process.terminate() self.stream = None def test_server(self): server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): assert_allclose(*data) assert_raises(StopIteration, next, server_data) def test_pickling(self): self.stream = cPickle.loads(cPickle.dumps(self.stream)) server_data = self.stream.get_epoch_iterator() expected_data = get_stream().get_epoch_iterator() for _, s, e in zip(range(3), server_data, expected_data): for data in zip(s, e): assert_allclose(*data, rtol=1e-5) assert_raises(StopIteration, next, server_data) def test_value_error_on_request(self): assert_raises(ValueError, self.stream.get_data, [0, 1]) def test_close(self): self.stream.close() def test_next_epoch(self): self.stream.next_epoch() def test_reset(self): self.stream.reset()