add a SkipOnCI trait fore tests

This commit is contained in:
mvenditto 2023-06-15 19:58:03 +01:00 committed by AT
parent ecafacd268
commit 2cbe791e5c
3 changed files with 17 additions and 9 deletions

View File

@ -1,5 +1,5 @@
namespace Gpt4All.Tests namespace Gpt4All.Tests;
{
public static class Constants public static class Constants
{ {
public const string MODELS_BASE_DIR = "../../../models"; public const string MODELS_BASE_DIR = "../../../models";
@ -7,4 +7,3 @@
public const string GPTJ_MODEL_PATH = $"{MODELS_BASE_DIR}/ggml-gpt4all-j-v1.3-groovy.bin"; public const string GPTJ_MODEL_PATH = $"{MODELS_BASE_DIR}/ggml-gpt4all-j-v1.3-groovy.bin";
public const string MPT_MODEL_PATH = $"{MODELS_BASE_DIR}/ggml-mpt-7b-chat.bin"; public const string MPT_MODEL_PATH = $"{MODELS_BASE_DIR}/ggml-mpt-7b-chat.bin";
} }
}

View File

@ -1,4 +1,4 @@
using Xunit; using Xunit;
namespace Gpt4All.Tests; namespace Gpt4All.Tests;
@ -12,18 +12,21 @@ public class ModelFactoryTests
} }
[Fact] [Fact]
[Trait(Traits.SkipOnCI, "True")]
public void CanLoadLlamaModel() public void CanLoadLlamaModel()
{ {
using var model = _modelFactory.LoadModel(Constants.LLAMA_MODEL_PATH); using var model = _modelFactory.LoadModel(Constants.LLAMA_MODEL_PATH);
} }
[Fact] [Fact]
[Trait(Traits.SkipOnCI, "True")]
public void CanLoadGptjModel() public void CanLoadGptjModel()
{ {
using var model = _modelFactory.LoadModel(Constants.GPTJ_MODEL_PATH); using var model = _modelFactory.LoadModel(Constants.GPTJ_MODEL_PATH);
} }
[Fact] [Fact]
[Trait(Traits.SkipOnCI, "True")]
public void CanLoadMptModel() public void CanLoadMptModel()
{ {
using var model = _modelFactory.LoadModel(Constants.MPT_MODEL_PATH); using var model = _modelFactory.LoadModel(Constants.MPT_MODEL_PATH);

View File

@ -0,0 +1,6 @@
namespace Gpt4All.Tests;
public static class Traits
{
public const string SkipOnCI = "SKIP_ON_CI";
}