add todo for farrah in read input test

orbitize/example_data/reflected_light_example.csv

@@ -0,0 +1,10 @@
+epoch,object,raoff,decoff,raoff_err,decoff_err,brightness
+57298,1,253.72,92.35,2.98,2.85,0.5
+57606,1,236.63,127.94,9.77,9.18,0.5
+57645,1,234.52,123.39,1.79,1.03,0.5
+57946,1,210.76,152.09,1.94,1.88,0.5
+58276,1,167.49,180.87,1.61,16.97,0.5
+58287,1,177.67,174.6,1.67,1.67,0.5
+58365,1,165.7,185.33,3.28,3.66,0.5
+58368,1,170.38,185.94,2.52,2.74,0.5
+58414,1,161.64,176.21,13.6,14.31,

orbitize/read_input.py

@@ -292,7 +292,7 @@ def read_file(filename):
         if have_brightness[index]:
             brightness = row["brightness"]
         else:
-            brightness = None
+            brightness = np.nan
 
         # check that "object" is an integer (instead of ABC/bcd)
         if not isinstance(row["object"], (int, np.int32, np.int64)):

tests/test_brightness.py

@@ -40,7 +40,6 @@ def test_brightness_calculation():
     )
 
     ra, dec, vz, brightness = test_system.compute_all_orbits(params)
-    # TODO (farrah): make plot of brightness vs time
 
     plt.figure()
     plt.scatter(times, brightness)
@@ -51,20 +50,20 @@ def test_brightness_calculation():
 
 def test_read_input_with_brightness():
 
-    # TODO (farrah): use code above as inspiration to read in a csv file with a brightness column
     num_secondary_bodies = 1
 
-    # input_file = os.path.join(DATADIR, "GJ504.csv")
-    input_file = os.path.join(DATADIR, "betaPic.csv")
+    input_file = os.path.join(DATADIR, "reflected_light_example.csv")
 
     data_table = read_input.read_file(input_file)
 
     times = data_table["epoch"].value
     brightness_values = data_table["brightness"].value
 
-    # Do we need the rest of this? since the values for time and brightness are given
+    print(data_table)
 
-    print("hello! :D ")
+    # TODO (Farrah): add a test that asserts the brightness column of the data table is 
+    # what you expect (hint: check in the reflected_light_example.csv to see what
+    # the brightness values should be
 
 
 def test_compute_posteriors():
@@ -113,7 +112,5 @@ def test_compute_posteriors():
 
 if __name__ == "__main__":
     # test_brightness_calculation()
-    # test_read_input_with_brightness()
-    test_compute_posteriors()
-
-    # Test commit
+    test_read_input_with_brightness()
+    # test_compute_posteriors()