Skip to content

Commit 44c1d12

Browse files
dance858Transurgeon
authored andcommitted
one more test
1 parent c65e543 commit 44c1d12

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

cvxpy/tests/nlp_tests/stress_tests_diff_engine/test_matmul_sparse.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,3 +113,17 @@ def test_sparse_and_dense_matmul(self):
113113
problem.solve(nlp=True, verbose=True)
114114
checker = DerivativeChecker(problem)
115115
checker.run_and_assert()
116+
117+
def test_sparse_and_dense_matmul2(self):
118+
np.random.seed(0)
119+
m, n = 4, 3
120+
A = np.random.rand(n, m)
121+
C = sp.random(m, n, density=0.5)
122+
X = cp.Variable((n, n), nonneg=True)
123+
B = np.random.rand(m, m)
124+
obj = cp.Minimize(cp.sum_squares(C @ X @ A - B))
125+
constraints = []
126+
problem = cp.Problem(obj, constraints)
127+
problem.solve(nlp=True, verbose=True)
128+
checker = DerivativeChecker(problem)
129+
checker.run_and_assert()

0 commit comments

Comments
 (0)