Example #1
0
    def test_get_model_file(self):
        client = Client("key", "value")
        client.get_model_file(86, 24, include_file=False)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/", params={"file": 0})

        client._get.reset_mock()

        client.get_model_file(86, 24, include_file=True)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/", params={"file": 1})

        client._get.reset_mock()

        client.get_model_file(86, 24)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/", params={"file": 0})
Example #2
0
    def test_get_model_file(self):
        client = Client("key", "value")
        client.get_model_file(86, 24, include_file=False)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/",
                                       params={"file": 0})

        client._get.reset_mock()

        client.get_model_file(86, 24, include_file=True)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/",
                                       params={"file": 1})

        client._get.reset_mock()

        client.get_model_file(86, 24)
        client._get.assert_called()
        client._get.assert_called_with("/models/86/files/24/",
                                       params={"file": 0})