@@ -322,19 +322,19 @@ class TestSlaterKosterBond:
322322 """
323323
324324 @pytest .mark .parametrize (
325- 'orb1_symbol, orb2_symbol , bravais_vector, expected_name' ,
325+ 'orb1_l_number, orb2_l_number , bravais_vector, expected_name' ,
326326 [
327327 (None , None , None , None ),
328- ('s' , None , None , None ),
329- (None , 'p' , None , None ),
330- ('s' , 's' , (0 , 0 , 0 ), 'sss' ),
331- ('s' , 'p' , (0 , 0 , 0 ), 'sps' ),
328+ (0 , None , None , None ),
329+ (None , 1 , None , None ),
330+ (0 , 0 , (0 , 0 , 0 ), 'sss' ),
331+ (0 , 1 , (0 , 0 , 0 ), 'sps' ),
332332 ],
333333 )
334334 def test_resolve_bond_name_from_references (
335335 self ,
336- orb1_symbol : str | None ,
337- orb2_symbol : str | None ,
336+ orb1_l_number : int | None ,
337+ orb2_l_number : int | None ,
338338 bravais_vector : tuple | None ,
339339 expected_name : str | None ,
340340 ):
@@ -344,13 +344,13 @@ def test_resolve_bond_name_from_references(
344344 sk_bond = SlaterKosterBond ()
345345 # If there's an orbit1 or orbit2, build them
346346 orbit1 = None
347- if orb1_symbol :
348- spherical_state = SphericalSymmetryState (l_quantum_symbol = orb1_symbol )
347+ if orb1_l_number is not None :
348+ spherical_state = SphericalSymmetryState (l_quantum_number = orb1_l_number )
349349 orbit1 = ElectronicState (spin_orbit_state = spherical_state )
350350
351351 orbit2 = None
352- if orb2_symbol :
353- spherical_state = SphericalSymmetryState (l_quantum_symbol = orb2_symbol )
352+ if orb2_l_number is not None :
353+ spherical_state = SphericalSymmetryState (l_quantum_number = orb2_l_number )
354354 orbit2 = ElectronicState (spin_orbit_state = spherical_state )
355355
356356 name = sk_bond .resolve_bond_name_from_references (
@@ -362,17 +362,17 @@ def test_resolve_bond_name_from_references(
362362 assert name == expected_name
363363
364364 @pytest .mark .parametrize (
365- 'orb1_symbol, orb2_symbol , bravais_vector, expected' ,
365+ 'orb1_l_number, orb2_l_number , bravais_vector, expected' ,
366366 [
367367 (None , None , None , None ),
368- ('s' , None , None , None ),
369- ('s' , 'p' , (0 , 0 , 0 ), 'sps' ),
368+ (0 , None , None , None ),
369+ (0 , 1 , (0 , 0 , 0 ), 'sps' ),
370370 ],
371371 )
372372 def test_normalize (
373373 self ,
374- orb1_symbol : str | None ,
375- orb2_symbol : str | None ,
374+ orb1_l_number : int | None ,
375+ orb2_l_number : int | None ,
376376 bravais_vector : tuple | None ,
377377 expected : str | None ,
378378 ):
@@ -382,13 +382,13 @@ def test_normalize(
382382 # Prepare a model scenario
383383 bond = SlaterKosterBond ()
384384 orbitals = []
385- if orb1_symbol :
386- spherical_state = SphericalSymmetryState (l_quantum_symbol = orb1_symbol )
385+ if orb1_l_number is not None :
386+ spherical_state = SphericalSymmetryState (l_quantum_number = orb1_l_number )
387387 electronic_state = ElectronicState (spin_orbit_state = spherical_state )
388388 orbitals .append (electronic_state )
389389 bond .orbital_1 = orbitals [- 1 ]
390- if orb2_symbol :
391- spherical_state = SphericalSymmetryState (l_quantum_symbol = orb2_symbol )
390+ if orb2_l_number is not None :
391+ spherical_state = SphericalSymmetryState (l_quantum_number = orb2_l_number )
392392 electronic_state = ElectronicState (spin_orbit_state = spherical_state )
393393 orbitals .append (electronic_state )
394394 bond .orbital_2 = orbitals [- 1 ]
0 commit comments