def test_get_batch(self): dd = DeepmdData(self.data_name) data = dd.get_batch(5) self._comp_np_mat2(np.sort(data['coord'], axis=0), np.sort(self.coord_bar, axis=0)) data = dd.get_batch(5) self._comp_np_mat2(np.sort(data['coord'], axis=0), np.sort(self.coord, axis=0)) data = dd.get_batch(5) self._comp_np_mat2(np.sort(data['coord'], axis=0), np.sort(self.coord_bar, axis=0)) data = dd.get_batch(5) self._comp_np_mat2(np.sort(data['coord'], axis=0), np.sort(self.coord, axis=0))
def test_get_batch(self) : dd = DeepmdData(self.data_name) data = dd.get_batch(5)