import rml.machines


def test_machine_load_elements():
    lattice = rml.machines.get_elements(machine='SRI21', elemType='BPM')
    assert len(lattice) == 173

import rml.machines


def test_machine_load_elements():
    lattice = rml.machines.get_elements(machine='SRI21', elemType='BPM')
    assert len(lattice) == 173
    for element in lattice.get_elements():
        assert element.get_pv_name('readback')

Test to get different pv names for an element
Test to get different pv names for an element

  import rml.machines
  
  
  def test_machine_load_elements():
      lattice = rml.machines.get_elements(machine='SRI21', elemType='BPM')
      assert len(lattice) == 173
+     for element in lattice.get_elements():
+         assert element.get_pv_name('readback')
  
Test to get different pv names for an element
## Code Before:
import rml.machines


def test_machine_load_elements():
    lattice = rml.machines.get_elements(machine='SRI21', elemType='BPM')
    assert len(lattice) == 173

## Instruction:
Test to get different pv names for an element
## Code After:
import rml.machines


def test_machine_load_elements():
    lattice = rml.machines.get_elements(machine='SRI21', elemType='BPM')
    assert len(lattice) == 173
    for element in lattice.get_elements():
        assert element.get_pv_name('readback')
