Merge branch 'main' of https://github.com/emil28092005/Transportation_Problem

2024-11-09 18:22:05 +03:00
parent b2709492ac fe6d8a88fc
commit e599fdaec9
1 changed files with 31 additions and 0 deletions
@@ -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]