Skip to content

Adopt Swift Testing in PlaintextMatrixTests and ConversionTests #165

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
fboemer merged 3 commits into from
Mar 13, 2025
Merged

Adopt Swift Testing in PlaintextMatrixTests and ConversionTests #165

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d8db4d0
Adopt Swift Testing in PlaintextMatrixTests
fpseverino Mar 13, 2025
5efc088
Adopt Swift Testing in ConversionTests
fpseverino Mar 13, 2025
599320a
Merge branch 'main' into testing
fpseverino Mar 13, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
85 changes: 47 additions & 38 deletions Tests/PrivateNearestNeighborSearchProtobufTests/ConversionTests.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Copyright 2024 Apple Inc. and the Swift Homomorphic Encryption project authors
// Copyright 2024-2025 Apple Inc. and the Swift Homomorphic Encryption project authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
Expand All @@ -16,7 +16,7 @@
@testable import PrivateNearestNeighborSearch
import PrivateNearestNeighborSearchProtobuf

import XCTest
import Testing

func increasingData<T: ScalarType>(dimensions: MatrixDimensions, modulus: T) -> [[T]] {
(0..<dimensions.rowCount).map { rowIndex in
Expand All @@ -27,32 +27,36 @@ func increasingData<T: ScalarType>(dimensions: MatrixDimensions, modulus: T) ->
}
}

class ConversionTests: XCTestCase {
func testDistanceMetric() throws {
@Suite
struct ConversionTests {
@Test
func distanceMetric() throws {
for metric in DistanceMetric.allCases {
XCTAssertEqual(try metric.proto().native(), metric)
#expect(try metric.proto().native() == metric)
}
}

func testPacking() throws {
XCTAssertEqual(
try MatrixPacking.denseColumn.proto().native(),
MatrixPacking.denseColumn)
XCTAssertEqual(
try MatrixPacking.denseRow.proto().native(),
MatrixPacking.denseRow)
@Test
func packing() throws {
#expect(
try MatrixPacking.denseColumn.proto().native() ==
MatrixPacking.denseColumn)
#expect(
try MatrixPacking.denseRow.proto().native() ==
MatrixPacking.denseRow)
let bsgs = BabyStepGiantStep(vectorDimension: 128)
XCTAssertEqual(bsgs.proto().native(), bsgs)
#expect(bsgs.proto().native() == bsgs)

XCTAssertEqual(
#expect(
try MatrixPacking
.diagonal(babyStepGiantStep: bsgs)
.proto()
.native(),
MatrixPacking.diagonal(babyStepGiantStep: bsgs))
.native() ==
MatrixPacking.diagonal(babyStepGiantStep: bsgs))
}

func testClientAndServerConfig() throws {
@Test
func clientAndServerConfig() throws {
func runTest<Scheme: HeScheme>(_: Scheme.Type) throws {
let vectorDimension = 4
let clientConfig = try ClientConfig<Scheme>(
Expand All @@ -67,35 +71,37 @@ class ConversionTests: XCTestCase {
significantBitCounts: [15],
preferringSmall: true,
nttDegree: 8))
XCTAssertEqual(try clientConfig.proto().native(), clientConfig)
#expect(try clientConfig.proto().native() == clientConfig)

let serverConfig = ServerConfig<Scheme>(
clientConfig: clientConfig,
databasePacking: MatrixPacking
.diagonal(
babyStepGiantStep: BabyStepGiantStep(vectorDimension: vectorDimension)))
XCTAssertEqual(try serverConfig.proto().native(), serverConfig)
#expect(try serverConfig.proto().native() == serverConfig)
}

try runTest(Bfv<UInt32>.self)
try runTest(Bfv<UInt64>.self)
}

func testDatabase() throws {
@Test
func database() throws {
let rows = (0...10).map { rowIndex in
DatabaseRow(
entryId: rowIndex,
entryMetadata: rowIndex.littleEndianBytes,
vector: [Float(rowIndex)])
}
for row in rows {
XCTAssertEqual(row.proto().native(), row)
#expect(row.proto().native() == row)
}
let database = Database(rows: rows)
XCTAssertEqual(database.proto().native(), database)
#expect(database.proto().native() == database)
}

func testSerializedPlaintextMatrix() throws {
@Test
func serializedPlaintextMatrix() throws {
func runTest<Scheme: HeScheme>(_: Scheme.Type) throws {
let encryptionParameters = try EncryptionParameters<Scheme>(from: .insecure_n_8_logq_5x18_logt_5)
let context = try Context<Scheme>(encryptionParameters: encryptionParameters)
Expand All @@ -110,27 +116,28 @@ class ConversionTests: XCTestCase {
packing: .denseColumn,
values: scalars.flatMap { $0 })
let serialized = try plaintextMatrix.serialize()
XCTAssertEqual(try serialized.proto().native(), serialized)
#expect(try serialized.proto().native() == serialized)
let deserialized = try PlaintextMatrix(deserialize: serialized, context: context)
XCTAssertEqual(deserialized, plaintextMatrix)
#expect(deserialized == plaintextMatrix)

for moduliCount in 1..<encryptionParameters.coefficientModuli.count {
let evalPlaintextMatrix = try plaintextMatrix.convertToEvalFormat(moduliCount: moduliCount)
let serialized = try evalPlaintextMatrix.serialize()
XCTAssertEqual(try serialized.proto().native(), serialized)
#expect(try serialized.proto().native() == serialized)
let deserialized = try PlaintextMatrix(
deserialize: serialized,
context: context,
moduliCount: moduliCount)
XCTAssertEqual(deserialized, evalPlaintextMatrix)
#expect(deserialized == evalPlaintextMatrix)
}
}

try runTest(Bfv<UInt32>.self)
try runTest(Bfv<UInt64>.self)
}

func testSerializedCiphertextMatrix() throws {
@Test
func serializedCiphertextMatrix() throws {
func runTest<Scheme: HeScheme>(_: Scheme.Type) throws {
let encryptionParameters = try EncryptionParameters<Scheme>(from: .insecure_n_8_logq_5x18_logt_5)
let context = try Context<Scheme>(encryptionParameters: encryptionParameters)
Expand All @@ -150,18 +157,18 @@ class ConversionTests: XCTestCase {
let ciphertextMatrix = try plaintextMatrix.encrypt(using: secretKey)
let serialized = try ciphertextMatrix.serialize()
let serializedProto = try serialized.proto()
XCTAssertEqual(try serializedProto.native(), serialized)
#expect(try serializedProto.native() == serialized)
}
// Check Evaluation format
do {
let ciphertextMatrix = try plaintextMatrix.encrypt(using: secretKey)
let evalCiphertextMatrix = try ciphertextMatrix.convertToEvalFormat()
let serialized = try evalCiphertextMatrix.serialize()
XCTAssertEqual(try serialized.proto().native(), serialized)
#expect(try serialized.proto().native() == serialized)
let deserialized = try CiphertextMatrix<Scheme, Eval>(
deserialize: serialized,
context: context)
XCTAssertEqual(deserialized, evalCiphertextMatrix)
#expect(deserialized == evalCiphertextMatrix)
}
// Check serializeForDecryption
do {
Expand All @@ -174,20 +181,21 @@ class ConversionTests: XCTestCase {
let serializedProto = try serialized.proto()
let serializedSize = try serializedProto.serializedData().count

XCTAssertLessThan(serializedForDecryptionSize, serializedSize)
#expect(serializedForDecryptionSize < serializedSize)
let deserialized = try CiphertextMatrix<Scheme, Scheme.CanonicalCiphertextFormat>(
deserialize: serializedForDecryption,
context: context, moduliCount: 1)
let decrypted = try deserialized.decrypt(using: secretKey)
XCTAssertEqual(decrypted, plaintextMatrix)
#expect(decrypted == plaintextMatrix)
}
}

try runTest(Bfv<UInt32>.self)
try runTest(Bfv<UInt64>.self)
}

func testQuery() throws {
@Test
func query() throws {
func runTest<Scheme: HeScheme>(_: Scheme.Type) throws {
let encryptionParameters = try EncryptionParameters<Scheme>(from: .insecure_n_8_logq_5x18_logt_5)
let context = try Context<Scheme>(encryptionParameters: encryptionParameters)
Expand All @@ -208,13 +216,14 @@ class ConversionTests: XCTestCase {

let query = Query(ciphertextMatrices: ciphertextMatrices)
let roundtrip = try query.proto().native(context: context)
XCTAssertEqual(roundtrip, query)
#expect(roundtrip == query)
}
try runTest(Bfv<UInt32>.self)
try runTest(Bfv<UInt64>.self)
}

func testSerializedProcessedDatabase() throws {
@Test
func serializedProcessedDatabase() throws {
func runTest<Scheme: HeScheme>(_: Scheme.Type) throws {
let encryptionParameters = try EncryptionParameters<Scheme>(from: .insecure_n_8_logq_5x18_logt_5)
let vectorDimension = 4
Expand All @@ -226,7 +235,7 @@ class ConversionTests: XCTestCase {
vector: Array(repeating: Float(rowIndex), count: vectorDimension))
}
for row in rows {
XCTAssertEqual(row.proto().native(), row)
#expect(row.proto().native() == row)
}
let database = Database(rows: rows)

Expand All @@ -250,7 +259,7 @@ class ConversionTests: XCTestCase {

let processed = try database.process(config: serverConfig)
let serialized = try processed.serialize()
XCTAssertEqual(try serialized.proto().native(), serialized)
#expect(try serialized.proto().native() == serialized)
}
try runTest(Bfv<UInt32>.self)
try runTest(Bfv<UInt64>.self)
Expand Down
Loading