def test_generate_subsamples(self): table = Table(np.array([[3, 1, 1], [0, 3, 3]]), ["O1", "O2"], ["S1", "S2", "S3"]) actual_o1 = set() actual_o2 = set() for i, obs in zip(range(100), generate_subsamples(table, 3)): actual_o1.add(tuple(obs.data("O1", "observation"))) actual_o2.add(tuple(obs.data("O2", "observation"))) self.assertEqual(actual_o1, {(3, 0, 0), (3, 1, 0), (3, 0, 1), (3, 1, 1)}) self.assertEqual(actual_o2, {(0, 3, 3), (0, 2, 3), (0, 3, 2), (0, 2, 2)})
def test_generate_subsamples(self): table = Table(np.array([[3, 1, 1], [0, 3, 3]]), ['O1', 'O2'], ['S1', 'S2', 'S3']) actual_o1 = set() actual_o2 = set() for i, obs in zip(range(100), generate_subsamples(table, 3)): actual_o1.add(tuple(obs.data('O1', 'observation'))) actual_o2.add(tuple(obs.data('O2', 'observation'))) self.assertEqual(actual_o1, {(3, 0, 0), (3, 1, 0), (3, 0, 1), (3, 1, 1)}) self.assertEqual(actual_o2, {(0, 3, 3), (0, 2, 3), (0, 3, 2), (0, 2, 2)})
def test_generate_subsamples(self): table = Table(np.array([[3, 1, 1], [0, 3, 3]]), ['O1', 'O2'], ['S1', 'S2', 'S3']) actual_o1 = set() actual_o2 = set() for i, obs in zip(range(100), generate_subsamples(table, 3)): actual_o1.add(tuple(obs.data('O1', 'observation'))) actual_o2.add(tuple(obs.data('O2', 'observation'))) self.assertEqual(actual_o1, {(3, 0, 0), (3, 1, 0), (3, 0, 1), (3, 1, 1)}) self.assertEqual(actual_o2, {(0, 3, 3), (0, 2, 3), (0, 3, 2), (0, 2, 2)})