def testDevicePathToDeviceName(self): self.assertEqual( "/job:ps/replica:1/task:2/cpu:0", debug_data.device_path_to_device_name( debug_data.METADATA_FILE_PREFIX + debug_data.DEVICE_TAG + ",job_ps,replica_1,task_2,cpu_0"))
def testDevicePathToDeviceName(self): self.assertEqual( "/job:ps/replica:1/task:2/cpu:0", debug_data.device_path_to_device_name( debug_data.METADATA_FILE_PREFIX + debug_data.DEVICE_TAG + ",job_ps,replica_1,task_2,cpu_0"))