35 lines
1.2 KiB
C#
35 lines
1.2 KiB
C#
using FluentAssertions;
|
|
using PerfectNumbers.Core;
|
|
using PerfectNumbers.Core.Gpu;
|
|
using Xunit;
|
|
using UInt128 = System.UInt128;
|
|
|
|
namespace PerfectNumbers.Core.Tests;
|
|
|
|
public class MersenneNumberResidueGpuTesterTests
|
|
{
|
|
private static LastDigit GetLastDigit(ulong exponent) => (exponent & 3UL) == 3UL ? LastDigit.Seven : LastDigit.One;
|
|
|
|
[Theory]
|
|
[InlineData(false)]
|
|
[InlineData(true)]
|
|
[Trait("Category", "Fast")]
|
|
public void Scan_handles_various_prime_exponents(bool useGpuOrder)
|
|
{
|
|
var tester = new MersenneNumberResidueGpuTester(useGpuOrder);
|
|
|
|
RunCase(tester, 23UL, 2UL, expectedPrime: false);
|
|
RunCase(tester, 29UL, 37UL, expectedPrime: false);
|
|
RunCase(tester, 31UL, 1_000UL, expectedPrime: true);
|
|
RunCase(tester, 89UL, 1_001UL, expectedPrime: true);
|
|
RunCase(tester, 127UL, 1_001UL, expectedPrime: true);
|
|
}
|
|
|
|
private static void RunCase(MersenneNumberResidueGpuTester tester, ulong exponent, ulong maxK, bool expectedPrime)
|
|
{
|
|
bool isPrime = true;
|
|
tester.Scan(exponent, (UInt128)exponent << 1, GetLastDigit(exponent), (UInt128)maxK, ref isPrime);
|
|
isPrime.Should().Be(expectedPrime);
|
|
}
|
|
}
|