a8eeb46f0f10ae97c82c09003b2319529b7ce4d3,nn_dataflow/tests/unit_test/test_partition_scheme.py,TestPartitionScheme,test_coordinate,#TestPartitionScheme#,129

Before Change

    def test_coordinate(self):
        """ Get coordinate. """
        nr1 = NodeRegion(origin=PhyDim2(0, 0), dim=self.ps1.dim(),
                         type=NodeRegion.PROC)
        nr2 = NodeRegion(origin=PhyDim2(0, 0), dim=self.ps2.dim(),
                         type=NodeRegion.PROC)

After Change

    def test_coordinate(self):
        """ Get coordinate. """
        for ps, nr in zip([self.ps1, self.ps2], [self.nr1, self.nr2]):

            coord_list = [ps.coordinate(nr, pidx) for pidx in ps.gen_pidx()]

Instances