Files
cuda-toolkit/__tests__/links/get-links.test.ts
2025-06-01 18:11:38 +02:00

28 lines
1012 B
TypeScript

import { LinuxLinks } from '../../src/links/linux-links'
import { WindowsLinks } from '../../src/links/windows-links'
import { getLinks } from '../../src/links/get-links'
test.concurrent('getLinks gives a valid ILinks class', async () => {
try {
const links = await getLinks()
expect(
links instanceof LinuxLinks || links instanceof WindowsLinks
).toBeTruthy()
} catch (error) {
throw new Error(`Error getting links: ${error}`)
// Other OS
}
})
test.concurrent('getLinks return same versions in same order', async () => {
const linuxLinks = LinuxLinks.Instance.getAvailableLocalCudaVersions()
const windowsLinks = WindowsLinks.Instance.getAvailableLocalCudaVersions()
const windowsNetworkLinks =
WindowsLinks.Instance.getAvailableNetworkCudaVersions()
expect(linuxLinks.length).toBe(windowsLinks.length)
expect(windowsLinks.length).toBe(windowsNetworkLinks.length)
expect(linuxLinks).toEqual(windowsLinks)
expect(windowsLinks).toEqual(windowsNetworkLinks)
})