fixed tests (#106)

* fixed tests

* removed typo

* fix typo

Co-authored-by: Matt Sokoloff <msokoloff@labelbox.com>

Author: msokoloff1

labelbox/schema/label.py

@@ -31,6 +31,8 @@ def __init__(self, *args, **kwargs):
     agreement = Field.Float("agreement")
     benchmark_agreement = Field.Float("benchmark_agreement")
     is_benchmark_reference = Field.Boolean("is_benchmark_reference")
+    updated_at = Field.DateTime("updated_at")
+    created_at = Field.DateTime("created_at")
 
     project = Relationship.ToOne("Project")
     data_row = Relationship.ToOne("DataRow")

tests/integration/test_label.py

@@ -59,10 +59,11 @@ def test_label_filter_order(client, rand_gen):
     data_row_2 = dataset_2.create_data_row(row_data=IMG_URL)
 
     l1 = project.create_label(data_row=data_row_1, label="l1")
+    time.sleep(1)  #Ensure there is no race condition
     l2 = project.create_label(data_row=data_row_2, label="l2")
 
     # Labels are not visible in the project immediately.
-    time.sleep(10)
+    time.sleep(20)
 
     # Filtering supported on dataset
     assert set(project.labels()) == {l1, l2}
@@ -71,8 +72,8 @@ def test_label_filter_order(client, rand_gen):
     assert set(project.labels(datasets=[dataset_2])) == {l2}
     assert set(project.labels(datasets=[dataset_1, dataset_2])) == {l1, l2}
 
-    assert list(project.labels(order_by=Label.label.asc)) == [l1, l2]
-    assert list(project.labels(order_by=Label.label.desc)) == [l2, l1]
+    assert list(project.labels(order_by=Label.created_at.asc)) == [l1, l2]
+    assert list(project.labels(order_by=Label.created_at.desc)) == [l2, l1]
 
     dataset_1.delete()
     dataset_2.delete()