def matrix():
data = np.matrix(
[[np.inf, 29, 82, 46, 68, 52, 72, 42, 51, 55, 29, 74, 23, 72, 46],
[29, np.inf, 55, 46, 42, 43, 43, 23, 23, 31, 41, 51, 11, 52, 21],
[82, 55, np.inf, 68, 46, 55, 23, 43, 41, 29, 79, 21, 64, 31, 51],
[46, 46, 68, np.inf, 82, 15, 72, 31, 62, 42, 21, 51, 51, 43, 64],
[68, 42, 46, 82, np.inf, 74, 23, 52, 21, 46, 82, 58, 46, 65, 23],
[52, 43, 55, 15, 74, np.inf, 61, 23, 55, 31, 33, 37, 51, 29, 59],
[72, 43, 23, 72, 23, 61, np.inf, 42, 23, 31, 77, 37, 51, 46, 33],
[42, 23, 43, 31, 52, 23, 42, np.inf, 33, 15, 37, 33, 33, 31, 37],
[51, 23, 41, 62, 21, 55, 23, 33, np.inf, 29, 62, 46, 29, 51, 11],
[55, 31, 29, 42, 46, 31, 31, 15, 29, np.inf, 51, 21, 41, 23, 37],
[29, 41, 79, 21, 82, 33, 77, 37, 62, 51, np.inf, 65, 42, 59, 61],
[74, 51, 21, 51, 58, 37, 37, 33, 46, 21, 65, np.inf, 61, 11, 55],
[23, 11, 64, 51, 46, 51, 51, 33, 29, 41, 42, 61, np.inf, 62, 23],
[72, 52, 31, 43, 65, 29, 46, 31, 51, 23, 59, 11, 62, np.inf, 59],
[46, 21, 51, 64, 23, 59, 33, 37, 11, 37, 61, 55, 23, 59, np.inf]])
cities = [
'c1', 'c2', 'c3', 'c4', 'c5', 'c6', 'c7', 'c8', 'c9', 'c10', 'c11',
'c12', 'c13', 'c14', 'c15'
]
test_matrix = tsp.TspMatrix(data, row_labels=cities, col_labels=cities)
return test_matrix
def matrix():
data = np.matrix([[np.inf, 10, 8, 9, 7], [10, np.inf, 10, 5, 6],
[8, 10, np.inf, 8, 9], [9, 5, 8, np.inf, 6],
[7, 6, 9, 6, np.inf]])
cities = ['a', 'b', 'c', 'd', 'e']
test_matrix = tsp.TspMatrix(data, row_labels=cities, col_labels=cities)
return test_matrix
def tree():
data = np.matrix([[np.inf, 10, 8, 9, 7], [10, np.inf, 10, 5, 6],
[8, 10, np.inf, 8, 9], [9, 5, 8, np.inf, 6],
[7, 6, 9, 6, np.inf]])
cities = ['a', 'b', 'c', 'd', 'e']
test_matrix = tsp.TspMatrix(data, row_labels=cities, col_labels=cities)
test_tree = b_n_b.BranchAndBoundTree(test_matrix)
return test_tree
def test_sub_row_min(matrix):
data = np.matrix([[np.inf, 3, 1, 2, 0], [5, np.inf, 5, 0, 1],
[0, 2, np.inf, 0, 1], [4, 0, 3, np.inf, 1],
[1, 0, 3, 0, np.inf]])
citiy_names = ['a', 'b', 'c', 'd', 'e']
m = tsp.TspMatrix(data, citiy_names, citiy_names)
matrix.sub_row_min()
assert np.array_equal(m, matrix)
def test_sub_col_min(matrix):
data = np.matrix([[np.inf, 5, 0, 4, 1], [3, np.inf, 2, 0, 0],
[1, 5, np.inf, 3, 3], [2, 0, 0, np.inf, 0],
[0, 1, 1, 1, np.inf]])
citiy_names = ['a', 'b', 'c', 'd', 'e']
m = tsp.TspMatrix(data, citiy_names, citiy_names)
matrix.sub_col_min()
assert np.array_equal(m, matrix)
def test_twice_around_tree():
b1 = True
b2 = True
b3 = True
d1 = np.matrix(
[[np.inf, 29, 82, 46, 68, 52, 72, 42, 51, 55, 29, 74, 23, 72, 46],
[29, np.inf, 55, 46, 42, 43, 43, 23, 23, 31, 41, 51, 11, 52, 21],
[82, 55, np.inf, 68, 46, 55, 23, 43, 41, 29, 79, 21, 64, 31, 51],
[46, 46, 68, np.inf, 82, 15, 72, 31, 62, 42, 21, 51, 51, 43, 64],
[68, 42, 46, 82, np.inf, 74, 23, 52, 21, 46, 82, 58, 46, 65, 23],
[52, 43, 55, 15, 74, np.inf, 61, 23, 55, 31, 33, 37, 51, 29, 59],
[72, 43, 23, 72, 23, 61, np.inf, 42, 23, 31, 77, 37, 51, 46, 33],
[42, 23, 43, 31, 52, 23, 42, np.inf, 33, 15, 37, 33, 33, 31, 37],
[51, 23, 41, 62, 21, 55, 23, 33, np.inf, 29, 62, 46, 29, 51, 11],
[55, 31, 29, 42, 46, 31, 31, 15, 29, np.inf, 51, 21, 41, 23, 37],
[29, 41, 79, 21, 82, 33, 77, 37, 62, 51, np.inf, 65, 42, 59, 61],
[74, 51, 21, 51, 58, 37, 37, 33, 46, 21, 65, np.inf, 61, 11, 55],
[23, 11, 64, 51, 46, 51, 51, 33, 29, 41, 42, 61, np.inf, 62, 23],
[72, 52, 31, 43, 65, 29, 46, 31, 51, 23, 59, 11, 62, np.inf, 59],
[46, 21, 51, 64, 23, 59, 33, 37, 11, 37, 61, 55, 23, 59, np.inf]])
cities1 = [
'c1', 'c2', 'c3', 'c4', 'c5', 'c6', 'c7', 'c8', 'c9', 'c10', 'c11',
'c12', 'c13', 'c14', 'c15'
]
matrix1 = tsp.TspMatrix(d1, cities1, cities1)
optimal1 = 291.0
d2 = np.matrix(
[[
np.inf, 4727, 1205, 6363, 3657, 3130, 2414, 563, 463, 5654, 1713,
1604, 2368, 2201, 1290, 1004, 3833, 2258, 3419, 2267, 2957, 720,
1700, 5279, 2578, 6076, 3465, 2654, 3625, 3115, 1574, 3951, 1748,
2142, 6755, 2383, 3306, 1029, 3530, 825, 2188, 4820, 3489, 1947,
6835, 1542, 2379, 3744
],
[
4727, np.inf, 3588, 2012, 1842, 6977, 6501, 5187, 5028, 2327,
4148, 4723, 3635, 3125, 4907, 3930, 7463, 6338, 7243, 5105, 4043,
4022, 3677, 2863, 3106, 1850, 7173, 6630, 1204, 6814, 6001, 3447,
5253, 2656, 3123, 6274, 7183, 5622, 3085, 4564, 2756, 1591, 7027,
6186, 3472, 5461, 4390, 2088
],
[
1205, 3588, np.inf, 5163, 2458, 3678, 3071, 1742, 1444, 4462,
1184, 1520, 1498, 1103, 1501, 951, 4298, 2903, 3967, 2169, 2209,
652, 828, 4136, 1518, 4873, 3954, 3254, 2446, 3581, 2441,
2960, 1966, 950, 5564, 2916, 3878, 2035, 2482, 1027, 1395, 3617,
3891, 2686, 5661, 2023, 1867, 2560
],
[
6363, 2012, 5163, np.inf, 2799, 8064, 7727, 6878, 6581, 1402,
5366, 5946, 4679, 4378, 6225, 5709, 8417, 7578, 8296, 6135, 4802,
5707, 4982, 2322, 4178, 320, 8186, 7800, 2778, 7859, 7408, 3763,
6461, 4223, 1427, 7451, 8263, 7131, 3669, 6011, 4638, 1681, 7987,
7502, 1877, 6758, 5360, 2844
],
[
3657, 1842, 2458, 2799, np.inf, 5330, 4946, 4200, 3824, 2012,
2573, 3157, 1924, 1580, 3427, 3179, 5749, 4793, 5577, 3409, 2223,
3066, 2185, 1860, 1401, 2491, 5486, 5035, 894, 5141, 4611, 1669,
3677, 1590, 3113, 4682, 5533, 4352, 1252, 3227, 2426, 1169, 5313,
4706, 3241, 3962, 2651, 304
],
[
3130, 6977, 3678, 8064, 5330, np.inf, 743, 3209, 2670, 6929, 2831,
2266, 3407, 3854, 2178, 4076, 727, 881, 293, 1930, 3310, 3672,
3315, 6199, 3932, 7745, 365, 482, 5774, 261, 1659, 4513, 1746,
4431, 7910, 769, 207, 2225, 4435, 2681, 5053, 6384, 550, 1224,
7805, 1670, 2704, 5230
],
[
2414, 6501, 3071, 7727, 4946, 743, np.inf, 2468, 1952, 6673, 2380,
1795, 3051, 3405, 1604, 3382, 1469, 168, 1020, 1681, 3110, 2993,
2827, 6009, 3552, 7412, 1104, 267, 5300, 821, 916, 4348, 1270,
3890, 7698, 332, 900, 1484, 4185, 2049, 4415, 6051, 1219, 482,
7635, 1054, 2432, 4884
],
[
563, 5187, 1742, 6878, 4200, 3209, 2468, np.inf, 718, 6203, 2241,
2051, 2920, 2762, 1687, 1304, 3932, 2331, 3487, 2669, 3487, 1175,
2260, 5840, 3141, 6596, 3563, 2728, 4120, 3240, 1559, 4507, 2082,
2658, 7304, 2512, 3364, 985, 4091, 1319, 2544, 5358, 3632, 1987,
7391, 1785, 2879, 4296
],
[
463, 5028, 1444, 6581, 3824, 2670, 1952, 718, np.inf, 5789, 1602,
1343, 2330, 2291, 970, 1451, 3376, 1796, 2959, 1951, 2835, 1112,
1725, 5346, 2628, 6285, 3007, 2193, 3889, 2661, 1122, 3920, 1372,
2391, 6883, 1927, 2845, 611, 3543, 676, 2590, 4993, 3039, 1486,
6934, 1112, 2196, 3876
],
[
5654, 2327, 4462, 1402, 2012, 6929, 6673, 6203, 5789, np.inf,
4392, 4947, 3648, 3501, 5274, 5183, 7216, 6535, 7140, 5022, 3621,
5077, 4090, 922, 3207, 1131, 7014, 6714, 2437, 6707, 6477, 2476,
5432, 3599, 1102, 6376, 7121, 6284, 2497, 5160, 4318, 937, 6795,
6507, 1268, 5773, 4249, 1914
],
[
1713, 4148, 1184, 5366, 2573, 2831, 2380, 2241, 1602, 4392,
np.inf, 586, 766, 1029, 883, 2040, 3353, 2224, 3100, 1049, 1246,
1625, 503, 3841, 1196, 5054, 3042, 2488, 2945, 2676, 2087, 2331,
1114, 1650, 5459, 2132, 3037, 1958, 1997, 931, 2513, 3701, 2923,
2137, 5459, 1394, 711, 2534
],
[
1604, 4723, 1520, 5946, 3157, 2266, 1795, 2051, 1343, 4947, 586,
np.inf, 1299, 1612, 406, 2208, 2824, 1639, 2542, 694, 1586, 1767,
1050, 4357, 1770, 5633, 2498, 1907, 3520, 2128, 1558, 2778, 531,
2171, 6003, 1552, 2472, 1538, 2506, 791, 2912, 4277, 2403, 1564,
5983, 827, 892, 3109
],
[
2368, 3635, 1498, 4679, 1924, 3407, 3051, 2920, 2330, 3648, 766,
1299, np.inf, 646, 1642, 2446, 3840, 2905, 3655, 1488, 730, 2096,
697, 3076, 533, 4363, 3567, 3122, 2453, 3219, 2842, 1592, 1791,
1480, 4706, 2772, 3610, 2721, 1232, 1656, 2550, 3001, 3403, 2860,
4697, 2126, 756, 1836
],
[
2201, 3125, 1103, 4378, 1580, 3854, 3405, 2762, 2291, 3501, 1029,
1612, 646, np.inf, 1853, 2026, 4349, 3247, 4119, 1997, 1341, 1753,
606, 3078, 419, 4070, 4052, 3517, 1923, 3690, 3032, 1866, 2142,
838, 4593, 3161, 4060, 2788, 1380, 1663, 1932, 2736, 3915, 3138,
4647, 2395, 1351, 1592
],
[
1290, 4907, 1501, 6225, 3427, 2178, 1604, 1687, 970, 5274, 883,
406, 1642, 1853, np.inf, 2029, 2803, 1438, 2466, 986, 1987, 1593,
1253, 4716, 2072, 5915, 2454, 1764, 3710, 2082, 1204, 3164, 497,
2287, 6342, 1419, 2379, 1134, 2867, 554, 2885, 4569, 2405, 1289,
6338, 555, 1297, 3406
],
[
1004, 3930, 951, 5709, 3179, 4076, 3382, 1304, 1451, 5183, 2040,
2208, 2446, 2026, 2029, np.inf, 4759, 3220, 4368, 2900, 3151, 442,
1765, 4960, 2444, 5443, 4396, 3610, 2932, 4034, 2572, 3891, 2525,
1590, 6278, 3313, 4261, 2033, 3398, 1476, 1241, 4287, 4390, 2928,
6419, 2428, 2749, 3337
],
[
3833, 7463, 4298, 8417, 5749, 727, 1469, 3932, 3376, 7216, 3353,
2824, 3840, 4349, 2803, 4759, np.inf, 1601, 477, 2359, 3617, 4345,
3851, 6433, 4372, 8098, 370, 1206, 6267, 726, 2384, 4754, 2335,
4991, 8148, 1452, 609, 2949, 4752, 3331, 5687, 6746, 437, 1948,
8005, 2334, 3098, 5618
],
[
2258, 6338, 2903, 7578, 4793, 881, 168, 2331, 1796, 6535, 2224,
1639, 2905, 3247, 1438, 3220, 1601, np.inf, 1165, 1563, 2988,
2829, 2666, 5882, 3401, 7263, 1233, 399, 5138, 923, 794, 4227,
1117, 3724, 7565, 286, 1049, 1348, 4051, 1881, 4248, 5903, 1322,
355, 7508, 887, 2302, 4736
],
[
3419, 7243, 3967, 8296, 5577, 293, 1020, 3487, 2959, 7140, 3100,
2542, 3655, 4119, 2466, 4368, 477, 1165, np.inf, 2170, 3520, 3965,
3588, 6393, 4183, 7977, 202, 767, 6041, 438, 1932, 4706, 2027,
4711, 8107, 1061, 132, 2503, 4652, 2972, 5344, 6617, 486, 1501,
7989, 1962, 2939, 5469
],
[
2267, 5105, 2169, 6135, 3409, 1930, 1681, 2669, 1951, 5022, 1049,
694, 1488, 1997, 986, 2900, 2359, 1563, 2170, np.inf, 1430, 2460,
1547, 4333, 2019, 5817, 2079, 1694, 3910, 1733, 1813, 2668, 654,
2694, 6029, 1366, 2130, 1991, 2525, 1474, 3542, 4455, 1923, 1641,
5957, 1071, 777, 3302
],
[
2957, 4043, 2209, 4802, 2223, 3310, 3110, 3487, 2835, 3621, 1246,
1586, 730, 1341, 1987, 3151, 3617, 2988, 3520, 1430, np.inf, 2779,
1387, 2905, 1062, 4482, 3398, 3119, 2922, 3087, 3115, 1240, 1953,
2175, 4607, 2796, 3501, 3119, 1136, 2173, 3268, 3136, 3189, 3029,
4527, 2355, 711, 2042
],
[
720, 4022, 652, 5707, 3066, 3672, 2993, 1175, 1112, 5077, 1625,
1767, 2096, 1753, 1593, 442, 4345, 2829, 3965, 2460, 2779, np.inf,
1401, 4781, 2166, 5427, 3984, 3212, 2946, 3620, 2224, 3603, 2089,
1496, 6178, 2906, 3861, 1719, 3132, 1040, 1479, 4211, 3969, 2553,
6290, 2012, 2336, 3189
],
[
1700, 3677, 828, 4982, 2185, 3315, 2827, 2260, 1725, 4090, 503,
1050, 697, 606, 1253, 1765, 3851, 2666, 3588, 1547, 1387, 1401,
np.inf, 3621, 903, 4675, 3537, 2954, 2475, 3169, 2427, 2254, 1578,
1148, 5177, 2598, 3521, 2194, 1833, 1074, 2054, 3340, 3423, 2541,
5213, 1801, 1077, 2190
],
[
5279, 2863, 4136, 2322, 1860, 6199, 6009, 5840, 5346, 922, 3841,
4357, 3076, 3078, 4716, 4960, 6433, 5882, 6393, 4333, 2905, 4781,
3621, np.inf, 2718, 2042, 6254, 6024, 2569, 5966, 5913, 1687,
4807, 3384, 1716, 5699, 6384, 5787, 1852, 4687, 4285, 1272, 6022,
5892, 1629, 5178, 3581, 1639
],
[
2578, 3106, 1518, 4178, 1401, 3932, 3552, 3141, 2628, 3207, 1196,
1770, 533, 419, 2072, 2444, 4372, 3401, 4183, 2019, 1062, 2166,
903, 2718, np.inf, 3864, 4097, 3635, 1932, 3748, 3274, 1448, 2284,
1164, 4286, 3283, 4136, 3086, 967, 1973, 2285, 2507, 3935, 3331,
4312, 2589, 1284, 1340
],
[
6076, 1850, 4873, 320, 2491, 7745, 7412, 6596, 6285, 1131, 5054,
5633, 4363, 4070, 5915, 5443, 8098, 7263, 7977, 5817, 4482, 5427,
4675, 2042, 3864, np.inf, 7866, 7483, 2515, 7539, 7101, 3449,
6146, 3938, 1375, 7134, 7944, 6831, 3349, 5709, 4397, 1363, 7667,
7190, 1798, 6446, 5041, 2528
],
[
3465, 7173, 3954, 8186, 5486, 365, 1104, 3563, 3007, 7014, 3042,
2498, 3567, 4052, 2454, 4396, 370, 1233, 202, 2079, 3398, 3984,
3537, 6254, 4097, 7866, np.inf, 839, 5973, 374, 2019, 4569, 1996,
4669, 7970, 1085, 305, 2581, 4532, 2976, 5339, 6509, 287, 1581,
7844, 1974, 2838, 5369
],
[
2654, 6630, 3254, 7800, 5035, 482, 267, 2728, 2193, 6714, 2488,
1907, 3122, 3517, 1764, 3610, 1206, 399, 767, 1694, 3119, 3212,
2954, 6024, 3635, 7483, 839, np.inf, 5427, 558, 1181, 4349, 1377,
4044, 7723, 356, 653, 1744, 4218, 2241, 4614, 6121, 955, 743,
7644, 1231, 2465, 4957
],
[
3625, 1204, 2446, 2778, 894, 5774, 5300, 4120, 3889, 2437, 2945,
3520, 2453, 1923, 3710, 2932, 6267, 5138, 6041, 3910, 2922, 2946,
2475, 2569, 1932, 2515, 5973, 5427, np.inf, 5612, 4824, 2550,
4050, 1498, 3476, 5071, 5980, 4470, 2096, 3388, 1911, 1501, 5831,
4994, 3704, 4264, 3209, 1196
],
[
3115, 6814, 3581, 7859, 5141, 261, 821, 3240, 2661, 6707, 2676,
2128, 3219, 3690, 2082, 4034, 726, 923, 438, 1733, 3087, 3620,
3169, 5966, 3748, 7539, 374, 558, 5612, np.inf, 1716, 4280, 1624,
4298, 7679, 735, 420, 2263, 4216, 2606, 4967, 6179, 400, 1277,
7567, 1609, 2501, 5032
],
[
1574, 6001, 2441, 7408, 4611, 1659, 916, 1559, 1122, 6477, 2087,
1558, 2842, 3032, 1204, 2572, 2384, 794, 1932, 1813, 3115, 2224,
2427, 5913, 3274, 7101, 2019, 1181, 4824, 1716, np.inf, 4330,
1180, 3346, 7545, 1023, 1808, 578, 4062, 1438, 3693, 5763, 2115,
440, 7537, 763, 2404, 4603
],
[
3951, 3447, 2960, 3763, 1669, 4513, 4348, 4507, 3920, 2476, 2331,
2778, 1592, 1866, 3164, 3891, 4754, 4227, 4706, 2668, 1240, 3603,
2254, 1687, 1448, 3449, 4569, 4349, 2550, 4280, 4330, np.inf,
3184, 2510, 3402, 4031, 4698, 4281, 533, 3245, 3612, 2187, 4339,
4265, 3296, 3576, 1941, 1381
],
[
1748, 5253, 1966, 6461, 3677, 1746, 1270, 2082, 1372, 5432, 1114,
531, 1791, 2142, 497, 2525, 2335, 1117, 2027, 654, 1953, 2089,
1578, 4807, 2284, 6146, 1996, 1377, 4050, 1624, 1180, 3184,
np.inf, 2685, 6475, 1022, 1952, 1341, 2963, 1050, 3358, 4787,
1926, 1086, 6436, 422, 1244, 3619
],
[
2142, 2656, 950, 4223, 1590, 4431, 3890, 2658, 2391, 3599, 1650,
2171, 1480, 838, 2287, 1590, 4991, 3724, 4711, 2694, 2175, 1496,
1148, 3384, 1164, 3938, 4669, 4044, 1498, 4298, 3346, 2510, 2685,
np.inf, 4697, 3693, 4636, 2975, 1981, 1909, 1124, 2718, 4565,
3548, 4830, 2839, 2140, 1751
],
[
6755, 3123, 5564, 1427, 3113, 7910, 7698, 7304, 6883, 1102, 5459,
6003, 4706, 4593, 6342, 6278, 8148, 7565, 8107, 6029, 4607, 6178,
5177, 1716, 4286, 1375, 7970, 7723, 3476, 7679, 7545, 3402, 6475,
4697, np.inf, 7393, 8097, 7370, 3515, 6249, 5379, 2001, 7738,
7556, 461, 6829, 5267, 3013
],
[
2383, 6274, 2916, 7451, 4682, 769, 332, 2512, 1927, 6376, 2132,
1552, 2772, 3161, 1419, 3313, 1452, 286, 1061, 1366, 2796, 2906,
2598, 5699, 3283, 7134, 1085, 356, 5071, 735, 1023, 4031, 1022,
3693, 7393, np.inf, 965, 1542, 3883, 1913, 4286, 5772, 1121, 600,
7322, 902, 2128, 4608
],
[
3306, 7183, 3878, 8263, 5533, 207, 900, 3364, 2845, 7121, 3037,
2472, 3610, 4060, 2379, 4261, 609, 1049, 132, 2130, 3501, 3861,
3521, 6384, 4136, 7944, 305, 653, 5980, 420, 1808, 4698, 1952,
4636, 8097, 965, np.inf, 2380, 4629, 2877, 5250, 6583, 570, 1380,
7986, 1866, 2904, 5432
],
[
1029, 5622, 2035, 7131, 4352, 2225, 1484, 985, 611, 6284, 1958,
1538, 2721, 2788, 1134, 2033, 2949, 1348, 2503, 1991, 3119, 1719,
2194, 5787, 3086, 6831, 2581, 1744, 4470, 2263, 578, 4281, 1341,
2975, 7370, 1542, 2380, np.inf, 3952, 1127, 3197, 5518, 2658,
1002, 7395, 951, 2429, 4380
],
[
3530, 3085, 2482, 3669, 1252, 4435, 4185, 4091, 3543, 2497, 1997,
2506, 1232, 1380, 2867, 3398, 4752, 4051, 4652, 2525, 1136, 3132,
1833, 1852, 967, 3349, 4532, 4218, 2096, 4216, 4062, 533, 2963,
1981, 3515, 3883, 4629, 3952, np.inf, 2873, 3080, 2012, 4324,
4046, 3478, 3328, 1755, 1000
],
[
825, 4564, 1027, 6011, 3227, 2681, 2049, 1319, 676, 5160, 931,
791, 1656, 1663, 554, 1476, 3331, 1881, 2972, 1474, 2173, 1040,
1074, 4687, 1973, 5709, 2976, 2241, 3388, 2606, 1438, 3245, 1050,
1909, 6249, 1913, 2877, 1127, 2873, np.inf, 2374, 4392, 2943,
1659, 6285, 1012, 1563, 3254
],
[
2188, 2756, 1395, 4638, 2426, 5053, 4415, 2544, 2590, 4318, 2513,
2912, 2550, 1932, 2885, 1241, 5687, 4248, 5344, 3542, 3268, 1479,
2054, 4285, 2285, 4397, 5339, 4614, 1911, 4967, 3693, 3612, 3358,
1124, 5379, 4286, 5250, 3197, 3080, 2374, np.inf, 3386, 5284,
3997, 5585, 3386, 3125, 2664
],
[
4820, 1591, 3617, 1681, 1169, 6384, 6051, 5358, 4993, 937, 3701,
4277, 3001, 2736, 4569, 4287, 6746, 5903, 6617, 4455, 3136, 4211,
3340, 1272, 2507, 1363, 6509, 6121, 1501, 6179, 5763, 2187, 4787,
2718, 2001, 5772, 6583, 5518, 2012, 4392, 3386, np.inf, 6314,
5837, 2205, 5095, 3680, 1169
],
[
3489, 7027, 3891, 7987, 5313, 550, 1219, 3632, 3039, 6795, 2923,
2403, 3403, 3915, 2405, 4390, 437, 1322, 486, 1923, 3189, 3969,
3423, 6022, 3935, 7667, 287, 955, 5831, 400, 2115, 4339, 1926,
4565, 7738, 1121, 570, 2658, 4324, 2943, 5284, 6314, np.inf, 1676,
7603, 1964, 2662, 5184
],
[
1947, 6186, 2686, 7502, 4706, 1224, 482, 1987, 1486, 6507, 2137,
1564, 2860, 3138, 1289, 2928, 1948, 355, 1501, 1641, 3029, 2553,
2541, 5892, 3331, 7190, 1581, 743, 4994, 1277, 440, 4265, 1086,
3548, 7556, 600, 1380, 1002, 4046, 1659, 3997, 5837, 1676, np.inf,
7521, 744, 2325, 4670
],
[
6835, 3472, 5661, 1877, 3241, 7805, 7635, 7391, 6934, 1268, 5459,
5983, 4697, 4647, 6338, 6419, 8005, 7508, 7989, 5957, 4527, 6290,
5213, 1629, 4312, 1798, 7844, 7644, 3704, 7567, 7537, 3296, 6436,
4830, 461, 7322, 7986, 7395, 3478, 6285, 5585, 2205, 7603, 7521,
np.inf, 6805, 5208, 3102
],
[
1542, 5461, 2023, 6758, 3962, 1670, 1054, 1785, 1112, 5773, 1394,
827, 2126, 2395, 555, 2428, 2334, 887, 1962, 1071, 2355, 2012,
1801, 5178, 2589, 6446, 1974, 1231, 4264, 1609, 763, 3576, 422,
2839, 6829, 902, 1866, 951, 3328, 1012, 3386, 5095, 1964, 744,
6805, np.inf, 1644, 3928
],
[
2379, 4390, 1867, 5360, 2651, 2704, 2432, 2879, 2196, 4249, 711,
892, 756, 1351, 1297, 2749, 3098, 2302, 2939, 777, 711, 2336,
1077, 3581, 1284, 5041, 2838, 2465, 3209, 2501, 2404, 1941, 1244,
2140, 5267, 2128, 2904, 2429, 1755, 1563, 3125, 3680, 2662, 2325,
5208, 1644, np.inf, 2532
],
[
3744, 2088, 2560, 2844, 304, 5230, 4884, 4296, 3876, 1914, 2534,
3109, 1836, 1592, 3406, 3337, 5618, 4736, 5469, 3302, 2042, 3189,
2190, 1639, 1340, 2528, 5369, 4957, 1196, 5032, 4603, 1381, 3619,
1751, 3013, 4608, 5432, 4380, 1000, 3254, 2664, 1169, 5184, 4670,
3102, 3928, 2532, np.inf
]])
cities2 = [
'a1', 'a2', 'a3', 'a4', 'a5', 'a6', 'a7', 'a8', 'a9', 'a10', 'a11',
'a12', 'a13', 'a14', 'a15', 'a16', 'a17', 'a18', 'a19', 'a20', 'a21',
'a22', 'a23', 'a24', 'a25', 'a26', 'a27', 'a28', 'a29', 'a30', 'a31',
'a32', 'a33', 'a34', 'a35', 'a36', 'a37', 'a38', 'a39', 'a40', 'a41',
'a42', 'a43', 'a44', 'a45', 'a46', 'a47', 'a48'
]
matrix2 = tsp.TspMatrix(d2, cities2, cities2)
optimal2 = 33551.0
d3 = np.matrix([[np.inf, 10, 8, 9, 7], [10, np.inf, 10, 5, 6],
[8, 10, np.inf, 8, 9], [9, 5, 8, np.inf, 6],
[7, 6, 9, 6, np.inf]])
cities3 = ['a', 'b', 'c', 'd', 'e']
matrix3 = tsp.TspMatrix(d3, row_labels=cities3, col_labels=cities3)
optimal3 = 34.0
result1 = tat.twice_around_tree(matrix1)
result2 = tat.twice_around_tree(matrix2)
result3 = tat.twice_around_tree(matrix3)
if result1['cost'] / optimal1 > 2:
b1 = False
if result2['cost'] / optimal2 > 2:
b2 = False
if result3['cost'] / optimal3 > 2:
b3 = False
assert b1 and b2 and b3
