public void testSaveLoad() throws IOException {
final MemoryPrintStream mps = new MemoryPrintStream();
final ZeroRBuilder.ZeroRClassifier zero = new ZeroRBuilder.ZeroRClassifier(50, 79);
zero.save(new DataOutputStream(mps.outputStream()), null);
final DataInputStream dis = new DataInputStream(new ByteArrayInputStream(mps.toByteArray()));
assertEquals(MlPredictLoader.MlPredictType.ZERO_R.ordinal(), dis.readInt());
final ZeroRBuilder.ZeroRClassifier zerob = new ZeroRBuilder.ZeroRClassifier(dis);
assertEquals(zero, zerob);
}
private void checkLoadVersion(int version) throws IOException {
final InputStream is =
Resources.getResourceAsStream("com/rtg/ml/resources/testZeroRVersion_" + version);
try (final DataInputStream dis = new DataInputStream(is)) {
final int type = dis.readInt();
assertEquals(MlPredictLoader.MlPredictType.ZERO_R.ordinal(), type);
final ZeroRBuilder.ZeroRClassifier bs = new ZeroRBuilder.ZeroRClassifier(dis);
assertEquals(version, bs.mCurrentVersion);
final StringBuilder str = bs.toString(new StringBuilder(), "", null);
final String s = str.toString();
assertTrue(s.contains("808/1007"));
}
}
