def test_make_commtime_function(): data = {("a", "b"): {"intercept": 1, "slope": 1}, ("b", "a"): {"intercept": 0, "slope": 10}} commtime = make_commtime_function(data, known_shapes) assert commtime(dotxx, "a", "b") == 4 * 100 * 100 * 1 + 1 assert commtime(dotxx, "b", "a") == 4 * 100 * 100 * 10 + 0
def test_make_commtime_function_non_clique_network(): data = { ("a", "b"): {"intercept": 1, "slope": 1}, ("b", "a"): {"intercept": 0, "slope": 10}, ("b", "c"): {"intercept": 0, "slope": 10}, } commtime = make_commtime_function(data, known_shapes) assert commtime(dotxx, "a", "b") == 4 * 100 * 100 * 1 + 1 assert commtime(dotxx, "b", "a") == 4 * 100 * 100 * 10 + 0 assert commtime(dotxx, "c", "a") > 10000
def test_make_commtime_function(): data = { ('a', 'b'): { 'intercept': 1, 'slope': 1 }, ('b', 'a'): { 'intercept': 0, 'slope': 10 } } commtime = make_commtime_function(data, known_shapes) assert commtime(dotxx, 'a', 'b') == 4 * 100 * 100 * 1 + 1 assert commtime(dotxx, 'b', 'a') == 4 * 100 * 100 * 10 + 0