From cc1a7167405212ceb3e67bb5a2f74e57406ac4d6 Mon Sep 17 00:00:00 2001 From: Vincent Esche Date: Sun, 11 Aug 2019 16:21:22 +0200 Subject: [PATCH] Fixed unit tests for `Matrix.sum(_:axies:)` --- Tests/SurgeTests/MatrixTests.swift | 35 +++++++++++++++++++++++++++++- 1 file changed, 34 insertions(+), 1 deletion(-) diff --git a/Tests/SurgeTests/MatrixTests.swift b/Tests/SurgeTests/MatrixTests.swift index 1f8994f..fba7434 100644 --- a/Tests/SurgeTests/MatrixTests.swift +++ b/Tests/SurgeTests/MatrixTests.swift @@ -515,6 +515,23 @@ class MatrixTests: XCTestCase { XCTAssertEqual(actual, expected, accuracy: 1e-8) } + func test_sum_matrix_rows_double() { + typealias Scalar = Double + + let lhs: Matrix = [ + [1, 2, 3], + [4, 5, 6], + ] + + let actual = sum(lhs, axies: .row) + let expected: Matrix = [ + [6], + [15], + ] + + XCTAssertEqual(actual, expected, accuracy: 1e-5) + } + func test_sum_matrix_rows_float() { typealias Scalar = Float @@ -523,7 +540,7 @@ class MatrixTests: XCTestCase { [4, 5, 6], ] - let actual = sum(lhs, axies: .column) + let actual = sum(lhs, axies: .row) let expected: Matrix = [ [6], [15], @@ -532,6 +549,22 @@ class MatrixTests: XCTestCase { XCTAssertEqual(actual, expected, accuracy: 1e-5) } + func test_sum_matrix_columns_double() { + typealias Scalar = Double + + let lhs: Matrix = [ + [1, 2, 3], + [4, 5, 6], + ] + + let actual = sum(lhs, axies: .column) + let expected: Matrix = [ + [5, 7, 9], + ] + + XCTAssertEqual(actual, expected, accuracy: 1e-5) + } + func test_sum_matrix_columns_float() { typealias Scalar = Float