// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using System;
using System.Runtime.InteropServices;
 
namespace Microsoft.ML.TestFramework.Attributes
{
    /// <summary>
    /// A fact for tests requiring X86 or X64 environment.
    /// </summary>
    public sealed class X86X64FactAttribute : EnvironmentSpecificFactAttribute
    {
        public X86X64FactAttribute(string skipMessage) : base(skipMessage)
        {
        }
 
        /// <inheritdoc />
        protected override bool IsEnvironmentSupported()
        {
            return RuntimeInformation.ProcessArchitecture == Architecture.X86 || RuntimeInformation.ProcessArchitecture == Architecture.X64;
        }
    }
}