Test for new attrs in gen file

Author: ppinchuk

tests/test_gen_config.py

@@ -94,12 +94,10 @@ def test_gen_from_config(runner, tech, clear_loggers):
         # get reV 2.0 generation profiles from disk
         rev2_profiles = None
         flist = os.listdir(run_dir)
-        print(flist)
         for fname in flist:
             if fname.endswith('.h5'):
                 path = os.path.join(run_dir, fname)
                 with Outputs(path, 'r') as cf:
-
                     msg = 'cf_profile not written to disk'
                     assert 'cf_profile' in cf.datasets, msg
                     rev2_profiles = cf['cf_profile']
@@ -115,6 +113,11 @@ def test_gen_from_config(runner, tech, clear_loggers):
                         else:
                             assert output not in cf.datasets
 
+                    assert "run_directory" in cf.h5.attrs
+                    assert "generation_config" in cf.h5.attrs
+                    assert (json.loads(cf.h5.attrs["generation_config"])
+                            == config)
+
                 break
 
         if rev2_profiles is None:
@@ -130,9 +133,6 @@ def test_gen_from_config(runner, tech, clear_loggers):
         result = np.allclose(rev1_profiles, rev2_profiles,
                              rtol=RTOL, atol=ATOL)
 
-        print(rev1_profiles)
-        print(rev2_profiles)
-
         clear_loggers()
         msg = ('reV generation from config input failed for "{}" module!'
                .format(tech))