def test_listTo2D(self): # check empty, it should return the correct dimension still test_array = [] final_array = [[]] self.assertEqual( util.listTo2D( test_array, 5 ), final_array ) test_array = range(20) final_array = [[0,1,2,3,4],[5,6,7,8,9],[10,11,12,13,14],[15,16,17,18,19]] self.assertEqual( util.listTo2D(test_array, 5), final_array) # test the odd sized conversion too test_array = range(19) final_array = [[0,1,2,3,4],[5,6,7,8,9],[10,11,12,13,14],[15,16,17,18]] self.assertEqual( util.listTo2D(test_array, 5), final_array)
def _formatFragmentNames(self): fragnames = list() names = self._fragmentation.getFragmentNames() for i in range(1, len(names) + 1): s = "%5s%03i" % (names[i - 1], i) fragnames.append(s) return join2D(listTo2D(fragnames, 5), ', ', ",\n ")
def _formatFragmentNames(self): fragnames = list() names = self._fragmentation.getFragmentNames() for i in range(1, len(names) + 1): s = "%5s%03i" % (names[i-1], i) fragnames.append(s) return join2D(listTo2D(fragnames, 5), ', ', ",\n ")
def _getFMOLayer(self): if self._nlayers == 1 or self._central_fragment < 1: return "" layers = self._fragment_layers list2D = listTo2D(layers, 10, '%i') return " LAYER(1)=%s" % join2D(list2D, ',', ",\n ")
def _getFMOActfg(self): if self._nlayers == 1: return "" if len(self._active_fragments) == 0: return "" return " MODFD=1\n IACTFG(1)=%s" % (join2D(listTo2D([i+1 for i in self._active_fragments],5,'%i'),',',',\n '))
def _getFMOMplevl(self): return " MPLEVL(1)=%s" % (join2D(listTo2D([0 for i in range(self._nlayers)],10,'%i'),',',",\n"))
def _formatCharges(self): list2D = listTo2D(self._fragmentation.getFragmentCharges(), 10, '%3i') return join2D(list2D, ',', ",\n ")
def _getFMOActfg(self): if self._nlayers == 1: return "" if len(self._active_fragments) == 0: return "" return " MODFD=1\n IACTFG(1)=%s" % (join2D( listTo2D([i + 1 for i in self._active_fragments], 5, '%i'), ',', ',\n '))
def _getFMOMplevl(self): return " MPLEVL(1)=%s" % (join2D( listTo2D([0 for i in range(self._nlayers)], 10, '%i'), ',', ",\n"))