def test_gen_iso_cool_lists2(self): """Test that string cooling steps are correctly accepted """ isolistraw = "a, b, c,d" coollistraw = "1_d, 99 y" # Create placeholder for phtn_src file with NTF() as myNTF: myNTF.write(self.NTFcontents) myNTF.seek(0) # Goes to beginning isolist, coollist = s2s.gen_iso_cool_lists(isolistraw, \ coollistraw, myNTF.name) self.assertEqual(isolist, ["a", "b", "c", "d"]) self.assertEqual(coollist, ["1_d", "99 y"])
def test_gen_iso_cool_lists4(self): """Test getting string cooling times from phtn_src file w/out of order cooling numbers. """ isolistraw = "a, b, c, d" coollistraw = "3, 1, 0, 2" # Create placeholder for phtn_src file with NTF() as myNTF: myNTF.write(self.NTFcontents) myNTF.seek(0) # Goes to beginning isolist, coollist = s2s.gen_iso_cool_lists(isolistraw, \ coollistraw, myNTF.name) self.assertEqual(isolist, ["a", "b", "c", "d"]) self.assertEqual(coollist, ["shutdown", "1_d", "2 w", "99 y"])
def test_gen_iso_cool_lists1(self): """Test parsing of non-numeric entries in isotopes and cooling input. Varied spacing is intentional. """ isolistraw = "a, b, c,d" coollistraw = "1,2, 3, 4"#"x, y,z,1 z" # Create placeholder for phtn_src file with NTF() as myNTF: myNTF.write(self.NTFcontents) myNTF.seek(0) # Goes to beginning isolist, coollist = s2s.gen_iso_cool_lists(isolistraw, \ coollistraw, myNTF.name) self.assertEqual(isolist, ["a", "b", "c", "d"]) self.assertEqual(coollist, ["1_d", "2 w", "99 y", "whoops"])
def test_gen_iso_cool_lists7(self): """Test that 'all' results in all cooling steps being grabbed. """ isolistraw = "a, b, c, d" coollistraw = "all" # Create placeholder for phtn_src file with NTF() as myNTF: myNTF.write(self.NTFcontents) myNTF.seek(0) # Goes to beginning isolist, coollist = s2s.gen_iso_cool_lists(isolistraw, \ coollistraw, myNTF.name) self.assertEqual(isolist, ["a", "b", "c", "d"]) self.assertEqual(coollist, ["shutdown", "1_d", "2 w", "99 y", \ "whoops", "too far"])