diff --git a/main.py b/main.py index 92c18ad..6151643 100644 --- a/main.py +++ b/main.py @@ -338,6 +338,37 @@ def TEST_CASE_1(): return solve(S, C, D, NWExpected, VogelExpected, RussellExpected) +def TEST_CASE_2(): + print("----------------------RUNNING_TEST_CASE_2----------------------") + C = np.array([[5, 8, 6], + [4, 7, 9], + [3, 8, 5]], dtype=np.int64) + + S = np.array([20, 30, 25], dtype=np.int64) + + D = np.array([10, 25, 40], dtype=np.int64) + print_problem_statement(S, C, D) + + NWExpected = np.array([ + [10, 10, 0], + [0, 15, 15], + [0, 0, 25], + ], dtype=np.int64) + + VogelExpected = np.array([ + [0], + [0], + [0] # Заполнить + ], dtype=np.int64) + + RussellExpected = np.array([ + [5, 0, 15], + [5, 25, 0], + [0, 0, 25] + ], dtype=np.int64) + + return solve(S, C, D, NWExpected, VogelExpected, RussellExpected) + '''if __name__ == "__main__": tests = [TEST_CASE_1]