def test_11_cwd_pathlib(self):
jobspec_path = pathlib.PosixPath(
self.jobspec_dir) / "valid" / "basic_v1.yaml"
jobspec = Jobspec.from_yaml_file(jobspec_path)
cwd = pathlib.PosixPath("/tmp")
jobspec.cwd = cwd
self.assertEqual(jobspec.cwd, os.fspath(cwd))
def test_07_count(self):
jobspec_fname = os.path.join(self.jobspec_dir, "valid", "use_case_2.4.yaml")
jobspec = Jobspec.from_yaml_file(jobspec_fname)
count_dict = jobspec.resource_counts()
self.assertEqual(count_dict["node"], 1)
self.assertEqual(count_dict["slot"], 11)
self.assertEqual(count_dict["core"], 16)
self.assertEqual(count_dict["memory"], 64)
def test_06_iter(self):
jobspec_fname = os.path.join(self.jobspec_dir, "valid",
"use_case_2.4.yaml")
jobspec = Jobspec.from_yaml_file(jobspec_fname)
self.assertEqual(len(list(jobspec)), 7)
self.assertEqual(
len(list([x for x in jobspec if x["type"] == "core"])), 2)
self.assertEqual(
len(list([x for x in jobspec if x["type"] == "slot"])), 2)
def test_04_valid_construction(self):
for jobspec_filepath in glob(
os.path.join(self.jobspec_dir, "valid", "*.yaml")):
Jobspec.from_yaml_file(jobspec_filepath)
