def test_dataset_serialization(self):
sgf_files = list(load_data_sets.find_sgf_files(TEST_DIR))
positions_w_context = list(load_data_sets.get_positions_from_sgf(sgf_files[0]))
dataset = load_data_sets.DataSet.from_positions_w_context(positions_w_context)
dataset.write(TEMP_FILE_NAME)
recovered = load_data_sets.DataSet.read(TEMP_FILE_NAME)
self.assertEqual(dataset.is_test, recovered.is_test)
self.assertEqual(dataset.data_size, recovered.data_size)
self.assertEqual(dataset.board_size, recovered.board_size)
self.assertEqual(dataset.input_planes, recovered.input_planes)
self.assertEqual(dataset.is_test, recovered.is_test)
self.assertEqual(dataset.pos_features.shape, recovered.pos_features.shape)
self.assertEqual(dataset.next_moves.shape, recovered.next_moves.shape)
self.assertEqualNPArray(dataset.next_moves, recovered.next_moves)
self.assertEqualNPArray(dataset.pos_features, recovered.pos_features)
def test_dataset_serialization(self):
sgf_files = list(load_data_sets.find_sgf_files(TEST_DIR))
positions_w_context = list(load_data_sets.get_positions_from_sgf(sgf_files[0]))
dataset = load_data_sets.DataSet.from_positions_w_context(positions_w_context)
dataset.write(TEMP_FILE_NAME)
recovered = load_data_sets.DataSet.read(TEMP_FILE_NAME)
self.assertEqual(dataset.is_test, recovered.is_test)
self.assertEqual(dataset.data_size, recovered.data_size)
self.assertEqual(dataset.board_size, recovered.board_size)
self.assertEqual(dataset.input_planes, recovered.input_planes)
self.assertEqual(dataset.is_test, recovered.is_test)
self.assertEqual(dataset.pos_features.shape, recovered.pos_features.shape)
self.assertEqual(dataset.next_moves.shape, recovered.next_moves.shape)
self.assertEqualNPArray(dataset.next_moves, recovered.next_moves)
self.assertEqualNPArray(dataset.pos_features, recovered.pos_features)