private void TestTeeSinkCustomToken(int api)
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream);
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
SinkTokenizer sink = new SinkTokenizer();
TokenStream stream1 = new PartOfSpeechAnnotatingFilter(sink);
stream = new TeeTokenFilter(stream, sink);
stream = new PartOfSpeechAnnotatingFilter(stream);
switch (api)
{
case 0:
ConsumeStreamNewAPI(stream);
ConsumeStreamNewAPI(stream1);
break;
case 1:
ConsumeStreamOldAPI(stream);
ConsumeStreamOldAPI(stream1);
break;
case 2:
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamVeryOldAPI(stream1);
break;
}
}
private void TestCachingCustomToken(int api)
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream);
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
stream = new CachingTokenFilter(stream); // <- the caching is done before the annotating!
stream = new PartOfSpeechAnnotatingFilter(stream);
switch (api)
{
case 0:
ConsumeStreamNewAPI(stream);
ConsumeStreamNewAPI(stream);
break;
case 1:
ConsumeStreamOldAPI(stream);
ConsumeStreamOldAPI(stream);
break;
case 2:
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamVeryOldAPI(stream);
break;
case 3:
ConsumeStreamNewAPI(stream);
ConsumeStreamOldAPI(stream);
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamNewAPI(stream);
ConsumeStreamVeryOldAPI(stream);
break;
}
}
public virtual void TestOnlyNewAPI()
{
TokenStream.SetOnlyUseNewAPI(true);
try
{
// this should fail with UOE
try
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream); // <-- this one is evil!
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
while (stream.IncrementToken())
;
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue((typeof(PartOfSpeechTaggingFilter).FullName + " does not implement incrementToken() which is needed for onlyUseNewAPI.").Equals(uoe.Message));
}
// this should pass, as all core token streams support the new API
TokenStream stream2 = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream2 = new LowerCaseFilter(stream2);
stream2 = new StopFilter(stream2, stopwords);
while (stream2.IncrementToken())
;
// Test, if all attributes are implemented by their implementation, not Token/TokenWrapper
Assert.IsTrue(stream2.AddAttribute(typeof(TermAttribute)) is TermAttributeImpl, "TermAttribute is implemented by TermAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(OffsetAttribute)) is OffsetAttributeImpl, "OffsetAttribute is implemented by OffsetAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(Lucene.Net.Analysis.Tokenattributes.FlagsAttribute)) is FlagsAttributeImpl, "FlagsAttribute is implemented by FlagsAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(PayloadAttribute)) is PayloadAttributeImpl, "PayloadAttribute is implemented by PayloadAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(PositionIncrementAttribute)) is PositionIncrementAttributeImpl, "PositionIncrementAttribute is implemented by PositionIncrementAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(TypeAttribute)) is TypeAttributeImpl, "TypeAttribute is implemented by TypeAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(SenselessAttribute)) is SenselessAttributeImpl, "SenselessAttribute is not implemented by SenselessAttributeImpl");
// try to call old API, this should fail
try
{
stream2.Reset();
Token reusableToken = new Token();
while ((reusableToken = stream2.Next(reusableToken)) != null)
;
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue("This TokenStream only supports the new Attributes API.".Equals(uoe.Message));
}
try
{
stream2.Reset();
while (stream2.Next() != null)
;
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue("This TokenStream only supports the new Attributes API.".Equals(uoe.Message));
}
// Test if the wrapper API (onlyUseNewAPI==false) uses TokenWrapper
// as attribute instance.
// TokenWrapper encapsulates a Token instance that can be exchanged
// by another Token instance without changing the AttributeImpl instance
// itsself.
TokenStream.SetOnlyUseNewAPI(false);
stream2 = new WhitespaceTokenizer(new System.IO.StringReader(doc));
Assert.IsTrue(stream2.AddAttribute(typeof(TermAttribute)) is TokenWrapper, "TermAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(OffsetAttribute)) is TokenWrapper, "OffsetAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(Lucene.Net.Analysis.Tokenattributes.FlagsAttribute)) is TokenWrapper, "FlagsAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(PayloadAttribute)) is TokenWrapper, "PayloadAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(PositionIncrementAttribute)) is TokenWrapper, "PositionIncrementAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(TypeAttribute)) is TokenWrapper, "TypeAttribute is implemented by TokenWrapper");
// This one is not implemented by TokenWrapper:
Assert.IsTrue(stream2.AddAttribute(typeof(SenselessAttribute)) is SenselessAttributeImpl, "SenselessAttribute is not implemented by SenselessAttributeImpl");
}
finally
{
TokenStream.SetOnlyUseNewAPI(false);
}
}
private void TestCachingCustomToken(int api)
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream);
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
stream = new CachingTokenFilter(stream); // <- the caching is done before the annotating!
stream = new PartOfSpeechAnnotatingFilter(stream);
switch (api)
{
case 0:
ConsumeStreamNewAPI(stream);
ConsumeStreamNewAPI(stream);
break;
case 1:
ConsumeStreamOldAPI(stream);
ConsumeStreamOldAPI(stream);
break;
case 2:
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamVeryOldAPI(stream);
break;
case 3:
ConsumeStreamNewAPI(stream);
ConsumeStreamOldAPI(stream);
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamNewAPI(stream);
ConsumeStreamVeryOldAPI(stream);
break;
}
}
private void TestTeeSinkCustomToken(int api)
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream);
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
SinkTokenizer sink = new SinkTokenizer();
TokenStream stream1 = new PartOfSpeechAnnotatingFilter(sink);
stream = new TeeTokenFilter(stream, sink);
stream = new PartOfSpeechAnnotatingFilter(stream);
switch (api)
{
case 0:
ConsumeStreamNewAPI(stream);
ConsumeStreamNewAPI(stream1);
break;
case 1:
ConsumeStreamOldAPI(stream);
ConsumeStreamOldAPI(stream1);
break;
case 2:
ConsumeStreamVeryOldAPI(stream);
ConsumeStreamVeryOldAPI(stream1);
break;
}
}
public virtual void TestOnlyNewAPI()
{
TokenStream.SetOnlyUseNewAPI(true);
try
{
// this should fail with UOE
try
{
TokenStream stream = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream = new PartOfSpeechTaggingFilter(stream); // <-- this one is evil!
stream = new LowerCaseFilter(stream);
stream = new StopFilter(stream, stopwords);
while (stream.IncrementToken())
{
;
}
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue((typeof(PartOfSpeechTaggingFilter).FullName + " does not implement incrementToken() which is needed for onlyUseNewAPI.").Equals(uoe.Message));
}
// this should pass, as all core token streams support the new API
TokenStream stream2 = new WhitespaceTokenizer(new System.IO.StringReader(doc));
stream2 = new LowerCaseFilter(stream2);
stream2 = new StopFilter(stream2, stopwords);
while (stream2.IncrementToken())
{
;
}
// Test, if all attributes are implemented by their implementation, not Token/TokenWrapper
Assert.IsTrue(stream2.AddAttribute(typeof(TermAttribute)) is TermAttributeImpl, "TermAttribute is implemented by TermAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(OffsetAttribute)) is OffsetAttributeImpl, "OffsetAttribute is implemented by OffsetAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(Lucene.Net.Analysis.Tokenattributes.FlagsAttribute)) is FlagsAttributeImpl, "FlagsAttribute is implemented by FlagsAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(PayloadAttribute)) is PayloadAttributeImpl, "PayloadAttribute is implemented by PayloadAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(PositionIncrementAttribute)) is PositionIncrementAttributeImpl, "PositionIncrementAttribute is implemented by PositionIncrementAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(TypeAttribute)) is TypeAttributeImpl, "TypeAttribute is implemented by TypeAttributeImpl");
Assert.IsTrue(stream2.AddAttribute(typeof(SenselessAttribute)) is SenselessAttributeImpl, "SenselessAttribute is not implemented by SenselessAttributeImpl");
// try to call old API, this should fail
try
{
stream2.Reset();
Token reusableToken = new Token();
while ((reusableToken = stream2.Next(reusableToken)) != null)
{
;
}
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue("This TokenStream only supports the new Attributes API.".Equals(uoe.Message));
}
try
{
stream2.Reset();
while (stream2.Next() != null)
{
;
}
Assert.Fail("If only the new API is allowed, this should fail with an UOE");
}
catch (System.NotSupportedException uoe)
{
Assert.IsTrue("This TokenStream only supports the new Attributes API.".Equals(uoe.Message));
}
// Test if the wrapper API (onlyUseNewAPI==false) uses TokenWrapper
// as attribute instance.
// TokenWrapper encapsulates a Token instance that can be exchanged
// by another Token instance without changing the AttributeImpl instance
// itsself.
TokenStream.SetOnlyUseNewAPI(false);
stream2 = new WhitespaceTokenizer(new System.IO.StringReader(doc));
Assert.IsTrue(stream2.AddAttribute(typeof(TermAttribute)) is TokenWrapper, "TermAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(OffsetAttribute)) is TokenWrapper, "OffsetAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(Lucene.Net.Analysis.Tokenattributes.FlagsAttribute)) is TokenWrapper, "FlagsAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(PayloadAttribute)) is TokenWrapper, "PayloadAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(PositionIncrementAttribute)) is TokenWrapper, "PositionIncrementAttribute is implemented by TokenWrapper");
Assert.IsTrue(stream2.AddAttribute(typeof(TypeAttribute)) is TokenWrapper, "TypeAttribute is implemented by TokenWrapper");
// This one is not implemented by TokenWrapper:
Assert.IsTrue(stream2.AddAttribute(typeof(SenselessAttribute)) is SenselessAttributeImpl, "SenselessAttribute is not implemented by SenselessAttributeImpl");
}
finally
{
TokenStream.SetOnlyUseNewAPI(false);
}
}
