fix tests

allenai · Jun 14, 2024 · 0d62c44 · 0d62c44
1 parent 1a7c61f
commit 0d62c44
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/client_test.py b/tests/client_test.py
@@ -2,13 +2,16 @@
 from flaky import flaky
 
 from beaker import Beaker
+from beaker.config import InternalConfig
 
 
 @flaky  # this can fail if the request to GitHub fails
 def test_warn_for_newer_version(monkeypatch):
     import beaker.client
     import beaker.version
 
+    InternalConfig().save()
+
     monkeypatch.setattr(Beaker, "CLIENT_VERSION", "0.1.0")
     monkeypatch.setattr(beaker.client, "_LATEST_VERSION_CHECKED", False)
 

diff --git a/tests/data_model_test.py b/tests/data_model_test.py
@@ -93,10 +93,16 @@ def test_experiment_spec_validation():
 
 def test_snake_case_vs_lower_camel_case():
     for x in (DataSource(host_path="/tmp/foo"), DataSource(hostPath="/tmp/foo")):  # type: ignore
-        assert str(x) == "DataSource(beaker=None, host_path='/tmp/foo', result=None, secret=None)"
+        assert (
+            str(x)
+            == "DataSource(beaker=None, host_path='/tmp/foo', weka=None, result=None, secret=None)"
+        )
         assert x.host_path == "/tmp/foo"
         x.host_path = "/tmp/bar"
-        assert str(x) == "DataSource(beaker=None, host_path='/tmp/bar', result=None, secret=None)"
+        assert (
+            str(x)
+            == "DataSource(beaker=None, host_path='/tmp/bar', weka=None, result=None, secret=None)"
+        )
         assert x.to_json() == {"hostPath": "/tmp/bar"}