@@ -57,8 +57,8 @@ void testDefaultFactory() throws IOException {
57
57
ChunkerModel model = trainModel (ModelType .MAXENT , new ChunkerFactory ());
58
58
59
59
ChunkerFactory factory = model .getFactory ();
60
- Assertions .assertTrue ( factory .getContextGenerator () instanceof DefaultChunkerContextGenerator );
61
- Assertions .assertTrue ( factory .getSequenceValidator () instanceof DefaultChunkerSequenceValidator );
60
+ Assertions .assertInstanceOf ( DefaultChunkerContextGenerator . class , factory .getContextGenerator ());
61
+ Assertions .assertInstanceOf ( DefaultChunkerSequenceValidator . class , factory .getSequenceValidator ());
62
62
63
63
ByteArrayOutputStream out = new ByteArrayOutputStream ();
64
64
model .serialize (out );
@@ -67,8 +67,8 @@ void testDefaultFactory() throws IOException {
67
67
ChunkerModel fromSerialized = new ChunkerModel (in );
68
68
69
69
factory = fromSerialized .getFactory ();
70
- Assertions .assertTrue ( factory .getContextGenerator () instanceof DefaultChunkerContextGenerator );
71
- Assertions .assertTrue ( factory .getSequenceValidator () instanceof DefaultChunkerSequenceValidator );
70
+ Assertions .assertInstanceOf ( DefaultChunkerContextGenerator . class , factory .getContextGenerator ());
71
+ Assertions .assertInstanceOf ( DefaultChunkerSequenceValidator . class , factory .getSequenceValidator ());
72
72
}
73
73
74
74
@@ -78,10 +78,10 @@ void testDummyFactory() throws IOException {
78
78
ChunkerModel model = trainModel (ModelType .MAXENT , new DummyChunkerFactory ());
79
79
80
80
DummyChunkerFactory factory = (DummyChunkerFactory ) model .getFactory ();
81
- Assertions .assertTrue ( factory . getContextGenerator ()
82
- instanceof DummyChunkerFactory . DummyContextGenerator );
83
- Assertions .assertTrue ( factory . getSequenceValidator ()
84
- instanceof DummyChunkerFactory . DummySequenceValidator );
81
+ Assertions .assertInstanceOf ( DummyChunkerFactory . DummyContextGenerator . class ,
82
+ factory . getContextGenerator () );
83
+ Assertions .assertInstanceOf ( DummyChunkerFactory . DummySequenceValidator . class ,
84
+ factory . getSequenceValidator () );
85
85
86
86
87
87
ByteArrayOutputStream out = new ByteArrayOutputStream ();
@@ -91,10 +91,8 @@ void testDummyFactory() throws IOException {
91
91
ChunkerModel fromSerialized = new ChunkerModel (in );
92
92
93
93
factory = (DummyChunkerFactory ) fromSerialized .getFactory ();
94
- Assertions .assertTrue (factory .getContextGenerator ()
95
- instanceof DefaultChunkerContextGenerator );
96
- Assertions .assertTrue (factory .getSequenceValidator ()
97
- instanceof DefaultChunkerSequenceValidator );
94
+ Assertions .assertInstanceOf (DefaultChunkerContextGenerator .class , factory .getContextGenerator ());
95
+ Assertions .assertInstanceOf (DefaultChunkerSequenceValidator .class , factory .getSequenceValidator ());
98
96
99
97
100
98
ChunkerME chunker = new ChunkerME (model );
0 commit comments