def test_between_block_pairs(self): expected = set() block_pairs = itertools.combinations(self.blocks, 2) for left_block, right_block in block_pairs: for x in left_block: for y in right_block: expected.add(frozenset((x, y))) actual = phdlib.between_block_pairs(self.blocks) actual_set = set(frozenset(pair) for pair in actual) self.assertEqual(len(actual), len(actual_set)) self.assertEqual(actual_set, expected)
def test_returns_ndarray(self): result = phdlib.between_block_pairs(self.blocks) self.assertIsInstance(result, np.ndarray)